From 7306a4392a30e01d1adaad4bd9672409beee8caa Mon Sep 17 00:00:00 2001
From: Jay Shi <xiaojies@google.com>
Date: Mon, 11 Jan 2021 13:08:36 -0800
Subject: [PATCH] [tf.data] Rolls out the optimization `map_parallelization` as
 experiment to 5% of Borg jobs.

PiperOrigin-RevId: 351218900
Change-Id: Iad9c50673c49d6cd136ad7c1826ffe59035c877e
---
 tensorflow/core/kernels/data/optimize_dataset_op.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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<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); };