diff options
author | Jiri Simsa <jsimsa@google.com> | 2018-09-25 13:42:46 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 13:46:54 -0700 |
commit | 348478f642216cf3cbe1eb67b875252d8e6a6418 (patch) | |
tree | c4c7afd4283506b2c413e429cab02fc547e13481 /tensorflow/contrib/kafka | |
parent | 976fb3105312bb17accebcbca2ebae906bcf99fb (diff) |
[tf.data] Adding a private method for (recursively) tracking dataset inputs.
PiperOrigin-RevId: 214495925
Diffstat (limited to 'tensorflow/contrib/kafka')
-rw-r--r-- | tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py b/tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py index a1624614d1..7129f09e8b 100644 --- a/tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py +++ b/tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py @@ -17,15 +17,15 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function -from tensorflow.contrib.kafka.python.ops import kafka_op_loader # pylint: disable=unused-import from tensorflow.contrib.kafka.python.ops import gen_dataset_ops -from tensorflow.python.data.ops.dataset_ops import Dataset +from tensorflow.contrib.kafka.python.ops import kafka_op_loader # pylint: disable=unused-import +from tensorflow.python.data.ops import dataset_ops from tensorflow.python.framework import dtypes from tensorflow.python.framework import ops from tensorflow.python.framework import tensor_shape -class KafkaDataset(Dataset): +class KafkaDataset(dataset_ops.DatasetSource): """A Kafka Dataset that consumes the message. """ |