region->getParentOp()->emitWarning()
        << "shape inference did not reach stable state after " << max_iterations
        << " iterations";
  }
  return !changed;
}

static FailureOr<bool> InferShapeForFunction(ShapeInference& context,
                                             func::FuncOp func,
                                             int64_t max_iterations) {
  FailureOr<bool> failure_or_converged =