// 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 = [{