// Infers shape for dataset ops that have `M` input elements and `N`
  // arguments, and also propagates the shape to the specified function (called
  // only when function exists and has single use).
  bool InferShapeForDatasetOpCommon(Operation* op, FuncOp f,
                                    int64_t max_iterations);

  // Infers the shape of ops that create TensorList. Specifically,