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

PiperOrigin-RevId: 351218900 Change-Id: Iad9c50673c49d6cd136ad7c1826ffe59035c877e
2021-01-11 13:08:36 -08:00 · 2021-01-11 13:08:36 -08:00 · 7306a4392a
commit 7306a4392a
parent 6e2d5e6c55
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", 1}
+        {"map_parallelization", 5}
    };
    // clang-format on
    auto hash_func = [](const string& str) { return Hash64(str); };