// as possible. This is to optimize the op order for the group-by-dialect pass,
// which factors consecutive same-dialect ops into functions.
class OrderByDialectPass
    : public impl::OrderByDialectPassBase<OrderByDialectPass> {
 public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(OrderByDialectPass)
  void runOnOperation() override;
};

int DialectOrdering(Operation* predecessor, Operation* op) {

def DeviceIndexSelectorPass : Pass<"tf-device-index-selector", "mlir::func::FuncOp"> {
  let summary = "Fold tf.DeviceIndex to constant.";
  let constructor = "TF::CreateDeviceIndexSelectorPass()";
}

def OrderByDialectPass : Pass<"tf-order-by-dialect", "mlir::ModuleOp"> {
  let summary = "Reorders ops so ops of the same dialect are next to each other.";
  let description = [{
      Performs a reordering of ops so that