}
    if (changed) {
      rewriter.finalizeOpModification(f);
      return success();
    } else {
      rewriter.cancelOpModification(f);
      return failure();
    }
  }
};

class HandleIf : public OpRewritePattern<TF::IfOp> {
  // Optional-agnostic pattern that propagates types across the program.
  using OpRewritePattern::OpRewritePattern;

  LogicalResult adjustBranch(TF::IfOp ifop, func::FuncOp branch,