return dataset_if;
}

// Populates WhileRegionOp body which is replacing `reduce_dataset`.  Iterates
// `anonymous_iterator` with `dataset_types` and optional calls `reduce_func`.
void PopulateDatasetWhileBody(OpBuilder builder, ReduceDatasetOp reduce_dataset,
                              FuncOp reduce_func, WhileRegionOp dataset_while,
                              AnonymousIteratorV3Op anonymous_iterator,