num_submodules++;
    clustering_pipeline_status = RunTFXLABridge(
        submodule,
        [](OpPassManager &pm) {
          internal::AddReplicatedBridgeClusteringPipelinePasses(pm);
        },
        /*module_name=*/"", /*dump_prefix=*/"tf_xla_clustering_bridge_v1");
    if (num_submodules > 1) {
      return mlir::WalkResult::interrupt();
    }

namespace tf2xla {
namespace internal {

using mlir::OpPassManager;
using mlir::func::FuncOp;

// Adds replicated Bridge clustering pipeline passes to the given pass_manager.
// Does not run them.
void AddReplicatedBridgeClusteringPipelinePasses(OpPassManager& pm,
                                                 llvm::StringRef module_name) {
  // The following ops must be preserved regardless of reachability. Ideally,