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

PiperOrigin-RevId: 353570967
Change-Id: I91d264da2c6d5cc5cd9de63b87977a82c6af3299
This commit is contained in:
Jay Shi 2021-01-24 20:52:50 -08:00 committed by TensorFlower Gardener
parent 6e3575fecd
commit 7d5d7d6ecb

View File

@ -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", 20}
{"map_parallelization", 50}
};
// clang-format on
auto hash_func = [](const string& str) { return Hash64(str); };