}

      mhlo_op = rewriter.create<DstOpT>(loc, TypeRange(while_result_types),
                                        adaptor.getOperands());
    } else {
      mhlo_op = rewriter.create<DstOpT>(loc, op.getResultTypes(),
                                        adaptor.getOperands());
    }

    int64_t num_regions = op.getNumRegions();
    for (int64_t idx = 0; idx < num_regions; ++idx) {