}
  }

  LogicalResult matchAndRewrite(TF::IfOp ifop,
                                PatternRewriter& rewriter) const override {
    bool success =
        succeeded(adjustBranch(ifop, ifop.then_function(), rewriter)) ||
        succeeded(adjustBranch(ifop, ifop.else_function(), rewriter));
    return LogicalResult::success(success);
  }
};

class HandleFunc : public OpRewritePattern<func::FuncOp> {
 public: