LogicalResult matchAndRewrite(
      mhlo::IfOp op, OpAdaptor adaptor,
      ConversionPatternRewriter& rewriter) const final {
    // HLO IfOp currently doesn't support stateless
    auto new_op = rewriter.create<TF::IfRegionOp>(
        op.getLoc(), op->getResultTypes(), op.getPred(),
        /*is_stateless=*/false, /*_then_func_name=*/nullptr,
        /*_else_func_name=*/nullptr);
    new_op.getThenBranch().takeBody(op.getTrueBranch());

    ConvertSigmoidGradOpDynamic,
    ConvertConv2DDynamic,
    ConvertPadOpDynamic,
    ConvertGatherNdOpDynamic,
    LowerControlFlowOp<TF::CaseRegionOp, mhlo::CaseOp>,
    LowerControlFlowOp<TF::IfRegionOp, mhlo::IfOp>,
    LowerControlFlowOp<TF::WhileRegionOp, mhlo::WhileOp>,
    LowerYieldOp>(context);
  // clang-format on
}
// LINT.ThenChange(:MlirAlwaysOps)
}  // end namespace mhlo