block.push_back(op);
          }
        }
      }
    });
  }
}

}  // namespace

std::unique_ptr<Pass> CreateOrderByDialectPass() {
  return std::make_unique<OrderByDialectPass>();
}

void RegisterOrderByDialectPass() { registerPass(CreateOrderByDialectPass); }

}  // namespace TF

#include "mlir/Pass/Pass.h"  // from @llvm-project

namespace mlir {
namespace TF {

// Create an instance of a pass that reorders ops so ops of the same dialect are
// next to each other.
std::unique_ptr<Pass> CreateOrderByDialectPass();

// Register this pass in the global registry of MLIR.
void RegisterOrderByDialectPass();

}  // namespace TF
}  // namespace mlir

  pm.addPass(mlir::createCanonicalizerPass());
  pm.addPass(mlir::createInlinerPass());
  pm.addPass(mlir::createSymbolDCEPass());
  pm.addPass(mlir::createCanonicalizerPass());

  pm.addPass(mlir::TF::CreateOrderByDialectPass());

  pm.addPass(mlir::mhlo::createHloLegalizeToStablehloPass());
}

// non-quantized inputs and outputs.
std::unique_ptr<OperationPass<func::FuncOp>> CreateLowerQuantizedPass();

// Reorders ops so ops of the same dialect are next to each other.
std::unique_ptr<Pass> CreateOrderByDialectPass();

// Groups ops into functions that only contain one dialect.
std::unique_ptr<Pass> CreateGroupByDialectPass();

// Removes unused parameters from functions & their callers.

      to
        %a = "x.f"()
        %c = "x.f"(%a)
        %b = "y.f"(%a)
      so that the two "x" dialect instructions are next to each other.
  }];
  let constructor = "TF::CreateOrderByDialectPass()";
}

def GroupByDialectPass : Pass<"tf-group-by-dialect", "mlir::ModuleOp"> {
  let summary = "Groups ops into functions that only contain one dialect.";
  let description = [{