diff options
Diffstat (limited to 'tensorflow/core/kernels/reader_dataset_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/reader_dataset_ops.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/reader_dataset_ops.cc b/tensorflow/core/kernels/reader_dataset_ops.cc index c08e42be1d..39ef92a5de 100644 --- a/tensorflow/core/kernels/reader_dataset_ops.cc +++ b/tensorflow/core/kernels/reader_dataset_ops.cc @@ -402,6 +402,7 @@ class FixedLengthRecordDatasetOp : public DatasetOpKernel { // Iteration ends when there are no more files to process. if (current_file_index_ == dataset()->filenames_.size()) { *end_of_sequence = true; + is_exhausted_ = true; return Status::OK(); } |