Home
last modified time | relevance | path

Searched refs:_TFRecordDataset (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dsnapshot_test.py122 dataset = core_readers._TFRecordDataset(filenames)
132 dataset2 = core_readers._TFRecordDataset(filenames)
146 dataset = core_readers._TFRecordDataset(filenames)
152 dataset2 = core_readers._TFRecordDataset(filenames)
167 dataset = core_readers._TFRecordDataset(filenames)
178 dataset2 = core_readers._TFRecordDataset(filenames)
193 dataset = core_readers._TFRecordDataset(filenames)
210 dataset2 = core_readers._TFRecordDataset(filenames)
686 dataset = core_readers._TFRecordDataset(filenames)
694 dataset2 = core_readers._TFRecordDataset(filenames)
[all …]
Dauto_shard_dataset_test.py350 dataset = core_readers._TFRecordDataset(self.test_filenames)
369 dataset = core_readers._TFRecordDataset(self.test_filenames)
409 dataset = core_readers._TFRecordDataset(self.test_filenames)
423 dataset = core_readers._TFRecordDataset(self.test_filenames)
/external/tensorflow/tensorflow/python/tpu/
Ddatasets.py39 def _TFRecordDataset(filename: Text) -> dataset_ops.Dataset: function
46 'tfrecord': _TFRecordDataset,
/external/tensorflow/tensorflow/python/data/ops/
Dreaders.py215 class _TFRecordDataset(dataset_ops.DatasetSource): class
241 super(_TFRecordDataset, self).__init__(variant_tensor)
343 return _TFRecordDataset(filename, compression_type, buffer_size)