From 7d5d7d6ecb5aee76cb720a6518f2981872c225bc Mon Sep 17 00:00:00 2001 From: Jay Shi Date: Sun, 24 Jan 2021 20:52:50 -0800 Subject: [PATCH] [tf.data] Rolls out the optimization `map_parallelization` as experiment to 50% of Borg jobs. PiperOrigin-RevId: 353570967 Change-Id: I91d264da2c6d5cc5cd9de63b87977a82c6af3299 --- 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 73a4d9e2afe..0606ca24da0 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", 20} + {"map_parallelization", 50} }; // clang-format on auto hash_func = [](const string& str) { return Hash64(str); };