// duplicate constants. Add canonicalizer to deduplicate.
  pm.addNestedPass<func::FuncOp>(mlir::createCanonicalizerPass());
  pm.addPass(TF::CreateXlaCallModuleSerializationPass());
}

void AddProcessNchwTensorPasses(OpPassManager& pm) {
  // For models with NCHW convolution format. This pass is required because

      return signalPassFailure();
    }

    RemoveSerializedStablehloFunctions(module);
  }
};

}  // namespace

std::unique_ptr<OperationPass<ModuleOp>>
CreateXlaCallModuleSerializationPass() {
  return std::make_unique<XlaCallModuleSerializationPass>();
}

}  // namespace TF

// Creates a pass that serializes StableHLO functions referenced by
// `tf.XlaCallModule` from the top-level module to `tf.XlaCallModule`'s
// `module` attribute.
std::unique_ptr<OperationPass<ModuleOp>> CreateXlaCallModuleSerializationPass();

}  // namespace TF

namespace tf_executor {

// Creates a pass to chain control outputs of while loop body.
std::unique_ptr<OperationPass<ModuleOp>>

    from the top-level module.
  }];

  let dependentDialects = [
    "stablehlo::StablehloDialect",
    "vhlo::VhloDialect",
  ];

  let constructor = "TF::CreateXlaCallModuleSerializationPass()";
}

def ColocateTPUCopyWithDynamicShapePass : Pass<"tf-colocate-tpu-copy-with-dynamic-shape", "ModuleOp"> {
  let summary = "Adjusts the device annotation of TPUCopyWithDynamicShape ops";