experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[tf.data] Add comment to ShouldWait for the nondeterministic case to explain the condition it's checking for

Browse Source 
PiperOrigin-RevId: 311032271
Change-Id: Ib83e68934898d22b810332a71e5efe142e62e139
This commit is contained in:
Rachel Lim 2020-05-11 17:55:13 -07:00 committed by TensorFlower Gardener
parent 775d6b6497
commit 9f7d843e3f
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/kernels/data/parallel_map_dataset_op.cc
View File

@ -621,6 +621,11 @@ class ParallelMapIterator : public DatasetBaseIterator {
return false;
}
if (!deterministic_) {
// Iterate through in-flight results and returns the first one that is
// found to be available and not end-of-input. If the first result (in
// order) is end-of-input, we know that all earlier iterations have
// already been completed, so it is safe to return that result for the
// caller to process end of iteration.
for (auto it = invocation_results_.begin();
it != invocation_results_.end(); ++it) {
if ((*it)->notification.HasBeenNotified() &&