// yields.
  bool InferShapeForIfRegion(IfRegionOp op);

  // Infers the shape CaseOp outputs based on the shapes of branch function
  // result types.
  bool InferShapeForCase(CaseOp op);

  // Infers the shape CaseRegion outputs based on the shapes of the branch
  // yields.
  bool InferShapeForCaseRegion(CaseRegionOp op);