[tf.data] Add `num_parallel_reads` argument to `tf.data.TFRecordDataset`.
authorDerek Murray <mrry@google.com>
Wed, 28 Feb 2018 22:36:09 +0000 (14:36 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Wed, 28 Feb 2018 22:40:41 +0000 (14:40 -0800)
commitb21969b1305b211cd08f8d628b6a5a0e7a9e16f8
tree457d92430b14ff31e0eb0e3b19deb33753618216
parent9f95084b53303af50d0a13fd9bb40a183af9104a
[tf.data] Add `num_parallel_reads` argument to `tf.data.TFRecordDataset`.

This provides a convenient way to use the `tf.contrib.data.parallel_interleave()` idiom for reading multiple TFRecord files in parallel. In addition, the `filenames` argument to the initializer can now be a `tf.data.Dataset` of strings, which makes it easier to use `TFRecordDataset` with `Dataset.list_files()`.

PiperOrigin-RevId: 187384812
tensorflow/contrib/data/python/ops/BUILD
tensorflow/contrib/data/python/ops/interleave_ops.py
tensorflow/python/data/kernel_tests/reader_dataset_ops_test.py
tensorflow/python/data/ops/BUILD
tensorflow/python/data/ops/readers.py
tensorflow/tools/api/golden/tensorflow.data.-t-f-record-dataset.pbtxt