};

TEST_F(SessionClusterTensorflowDialectTest, ClustersTf) {
  CellReader<int64_t> compilation_status(kCompilationStreamz);

  TF_ASSERT_OK(CreateMlirModule("empty_func.mlir"));

  TF_EXPECT_OK(
      RunSessionTf2xlaClusteringBridge(*mlir_module_,
                                       /*is_in_fallback_enabled_mode=*/false));
  EXPECT_EQ(compilation_status.Delta(mlir::TF::kMlirPh1BridgeCounterReplicated,

// input MLIR should not have infeed and outfeed ops, which are unsupported via
// this API. Returns OkStatus if passed, otherwise an error.
tensorflow::Status RunSessionTf2xlaClusteringBridge(
    mlir::ModuleOp module, bool is_in_fallback_enabled_mode);

}  // namespace v1
}  // namespace tf2xla
}  // namespace tensorflow

        /*error_prefix=*/"Bridge Errored running clustering pipeline:",
        is_in_fallback_enabled_mode, clustering_pipeline_status));
  }

  return absl::OkStatus();
}

tensorflow::Status RunSessionTf2xlaClusteringBridge(
    ModuleOp module, bool is_in_fallback_enabled_mode) {
  VLOG(2) << "TPU Sessions Bridge called stack trace is "
          << "(NOTE: this is not an error; rather the stack trace for "