From c32906c588162b631b0168763a8250b9ab669f82 Mon Sep 17 00:00:00 2001
From: Jay Shi <xiaojies@google.com>
Date: Mon, 12 Oct 2020 12:22:35 -0700
Subject: [PATCH] [tf.data] Increase the roll out percentage of optimization
 `enable_gradient_descent` to 5%.

PiperOrigin-RevId: 336717806
Change-Id: I724c31d9d2cb4cff1c867c168e52aa026c05c01f
---
 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 81fcf6d7b29..6e079937885 100644
--- a/tensorflow/core/kernels/data/optimize_dataset_op.cc
+++ b/tensorflow/core/kernels/data/optimize_dataset_op.cc
@@ -84,7 +84,7 @@ void OptimizeDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input,
     // of the Borg jobs, the experiments will be randomly turned on.
     // clang-format off
     absl::flat_hash_map<string, uint64> live_experiments = {
-        {"enable_gradient_descent", 1}
+        {"enable_gradient_descent", 5}
     };
     // clang-format on
     auto hash_func = [](const string& str) { return Hash64(str); };