TEST_F(FunctionClusterTensorflowDialectTest, ClustersTfReplicatedBridge) {
  CellReader<int64_t> compilation_status(kCompilationStreamz);

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

  TF_EXPECT_OK(RunFunctionTf2xlaClusteringBridge(
      *mlir_module_, /*is_supported_by_replicated_brige*/ true,
      /*is_in_fallback_enabled_mode=*/false));

  FuncOp main = mlir_module_->lookupSymbol<mlir::func::FuncOp>("main");
  ASSERT_TRUE(main);

//
// Output: Modifies the input module in place with clustered operations.
//   status - Whether the transformation to cluster the input MLIR module was
//   successful.
tensorflow::Status RunFunctionTf2xlaClusteringBridge(
    mlir::ModuleOp module, bool is_supported_by_replicated_brige,
    bool is_in_fallback_enabled_mode,
    llvm::StringRef module_name = llvm::StringRef());
}  // namespace v2
}  // namespace tf2xla

      pm, module_name);
}

void CreateReplicatedClusteringPipelineV2(OpPassManager &pm) {
  CreateReplicatedClusteringPipeline(pm, /*module_name=*/"");
}

tensorflow::Status RunFunctionTf2xlaClusteringBridge(
    ModuleOp module, bool is_supported_by_replicated_brige,
    bool is_in_fallback_enabled_mode, llvm::StringRef module_name) {
  std::string device_type = is_supported_by_replicated_brige