commit 0b6ae294694520b41b74ec8068a613344b24f485
author Wilsin Gosti <wilsin@google.com> Wed Jan 19 10:14:58 2022 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Wed Jan 19 10:22:05 2022 -0800
tree 13c8595b9d091cf1b0f7132d7be3524f1d2f5900
parent 860140d907a5d305c248c31732cf266eb065e0d0

[tf.data] If `parallelism` is set to `tf.data.AUTOTUNE` for an Op, sets its value to the smaller of 16 and number of schedulable CPUs when `GetNext()` is invoked on the Op. If `Autotune` is on, this number will be optimized to what `Autotune` thinks is the optimum value. If not, it stays at that value.

PiperOrigin-RevId: 422844422
Change-Id: Ic8ed01d5a6882a91fb3c996dd20e0121aebf1a68

tensorflow/core/data/dataset_utils.cc

diff --git a/tensorflow/core/data/dataset_utils.cc b/tensorflow/core/data/dataset_utils.cc
index c549654..afd23d2 100644
--- a/tensorflow/core/data/dataset_utils.cc
+++ b/tensorflow/core/data/dataset_utils.cc
@@ -889,7 +889,7 @@
 
 namespace {
 
-REGISTER_DATASET_EXPERIMENT("initial_parallelism_value", 0);
+REGISTER_DATASET_EXPERIMENT("initial_parallelism_value", 50);
 REGISTER_DATASET_EXPERIMENT("enable_bufferedio_v2", 100);
 REGISTER_DATASET_EXPERIMENT("inject_prefetch", 50);
 REGISTER_DATASET_EXPERIMENT("max_parallelism", 100);