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

[tf.data] Fix concurrency issue in experimental/parallel_interleave_dataset_op.cc.

Browse Source 
PiperOrigin-RevId: 269902894
This commit is contained in:
Rachel Lim 2019-09-18 15:21:17 -07:00 committed by TensorFlower Gardener
parent bdcf4b6296
commit 6274f037d4
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/kernels/data/experimental/parallel_interleave_dataset_op.cc
View File

@ -716,7 +716,11 @@ class ParallelInterleaveDatasetOp::Dataset : public DatasetBase {
// CHECKPOINT_MARKER_C
// Non-OK iterator creation status has been notified to the
// client.
workers_[thread_index].cond_var.notify_one();
if (dataset()->sloppy_) {
sloppy_cond_var_.notify_one();
} else {
workers_[thread_index].cond_var.notify_one();
}
} else {
bool end_of_sequence = false;
while (!end_of_sequence) {