diff options
author | Jiri Simsa <jsimsa@google.com> | 2018-05-31 13:43:43 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-31 13:45:55 -0700 |
commit | 89a55fef3316e0e270e0f87f71bd8c2d32443cc8 (patch) | |
tree | 20825ff4e2a98e83b3781c86751bc5d385d4f9fe /tensorflow/contrib/kafka | |
parent | b3adb58d84ebb91d893b647ab4081530460fb8ed (diff) |
[tf.data] Changing signature of `MakeIterator` to enable propagating error status.
PiperOrigin-RevId: 198772254
Diffstat (limited to 'tensorflow/contrib/kafka')
-rw-r--r-- | tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc b/tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc index a4cd4a2cc4..7b08cfa095 100644 --- a/tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc +++ b/tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc @@ -64,7 +64,7 @@ class KafkaDatasetOp : public DatasetOpKernel { eof_(eof), timeout_(timeout) {} - std::unique_ptr<IteratorBase> MakeIterator( + std::unique_ptr<IteratorBase> MakeIteratorInternal( const string& prefix) const override { return std::unique_ptr<IteratorBase>( new Iterator({this, strings::StrCat(prefix, "::Kafka")})); |