// new passes generally should go into the internal
  // AddReplicatedBridgeClusteringPipelinePasses.
  pm.addPass(mlir::TFTPU::CreateTPUValidateInputsPass());
  pm.addNestedPass<FuncOp>(
      mlir::TF::CreateCanonicalizeCompileAndReplicateAttributesPass());
  tensorflow::tf2xla::internal::AddReplicatedBridgeClusteringPipelinePasses(
      pm, module_name);
}

namespace tensorflow {
namespace tf2xla {
namespace internal {

using mlir::OpPassManager;

TEST(ClusteringBridgePassesTest, AddsBridgePasses) {
  OpPassManager pass_manager;
  AddReplicatedBridgeClusteringPipelinePasses(pass_manager);

  EXPECT_EQ(pass_manager.size(), 45);
}

TEST(ClusteringBridgePassesTest, AddsNonTPUBridgePasses) {
  OpPassManager pass_manager;