diff --git a/tensorflow/core/kernels/data/optimize_dataset_op.cc b/tensorflow/core/kernels/data/optimize_dataset_op.cc index d04da1359e5..4ea47936f48 100644 --- a/tensorflow/core/kernels/data/optimize_dataset_op.cc +++ b/tensorflow/core/kernels/data/optimize_dataset_op.cc @@ -85,7 +85,7 @@ void OptimizeDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, // clang-format off absl::flat_hash_map live_experiments = { {"enable_gradient_descent", 0}, - {"map_parallelization", 1} + {"map_parallelization", 5} }; // clang-format on auto hash_func = [](const string& str) { return Hash64(str); };