[tf.data] Rolls out the optimization map_parallelization as experiment to 1% of Borg jobs.

PiperOrigin-RevId: 350163563 Change-Id: I5e842ad12d10eca205a2e50027e96965197203ce
2021-01-05 10:03:43 -08:00 · 2021-01-05 10:03:43 -08:00 · 242f42a186
commit 242f42a186
parent a19a2cd6af
1 changed files with 1 additions and 1 deletions
--- 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<string, uint64> live_experiments = {
        {"enable_gradient_descent", 0},
-        {"map_parallelization", 0}
+        {"map_parallelization", 1}
    };
    // clang-format on
    auto hash_func = [](const string& str) { return Hash64(str); };