| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | make_tf_record_dataset_test.py | 35 num_parallel_reads=1, drop_final_batch=False, parser_fn=False): argument
52 num_parallel_reads=num_parallel_reads,
60 interleave_cycle_length=num_parallel_reads,
79 self._read_test(batch_size, num_epochs, num_parallel_reads=8)
91 self._read_test(batch_size, num_epochs, num_parallel_reads=8,
100 self._read_test(batch_size, num_epochs, num_parallel_reads=8,
103 def _shuffle_test(self, batch_size, num_epochs, num_parallel_reads=1, argument
109 num_parallel_reads=num_parallel_reads,
149 for num_parallel_reads in [1, 2]:
151 self._shuffle_test(batch_size, num_epochs, num_parallel_reads)
[all …]
|
| D | auto_shard_dataset_test.py | 144 num_parallel_reads=parallel_reads,
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | readers.py | 178 num_parallel_reads=None): argument
216 self._num_parallel_reads = num_parallel_reads
221 if num_parallel_reads is None:
225 filenames, read_one_file, cycle_length=num_parallel_reads,
235 num_parallel_reads=None): argument
239 num_parallel_reads or self._num_parallel_reads)
254 num_parallel_reads=None): argument
256 filenames, compression_type, buffer_size, num_parallel_reads)
264 num_parallel_reads=None): argument
270 num_parallel_reads or self._dataset._num_parallel_reads)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | datasets.py | 57 num_parallel_reads=None, argument
125 num_parallel_reads = num_parallel_reads or 8
147 reader_fn, cycle_length=num_parallel_reads, sloppy=sloppy))
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 228 num_parallel_reads=None, argument
275 if num_parallel_reads is None:
280 num_parallel_reads = 24
282 files, num_parallel_reads=num_parallel_reads)
328 num_parallel_reads=1, argument
499 filename_to_dataset, cycle_length=num_parallel_reads, sloppy=sloppy))
536 num_parallel_reads=1, argument
546 prefetch_buffer_size, num_parallel_reads, sloppy, num_rows_for_inference,
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | readers.py | 49 num_parallel_reads=1, argument
141 prefetch_buffer_size, num_parallel_reads, sloppy, num_rows_for_inference,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.-t-f-record-dataset.pbtxt | 8 …[\'self\', \'filenames\', \'compression_type\', \'buffer_size\', \'num_parallel_reads\'], varargs=…
|
| D | tensorflow.data.experimental.pbtxt | 145 …uffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_parallel_reads\', \'sloppy\…
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | tf_record_dataset_test.py | 165 dataset = readers.TFRecordDataset(files, num_parallel_reads=4)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.-t-f-record-dataset.pbtxt | 22 …[\'self\', \'filenames\', \'compression_type\', \'buffer_size\', \'num_parallel_reads\'], varargs=…
|
| D | tensorflow.data.experimental.pbtxt | 145 …uffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_parallel_reads\', \'sloppy\…
|
| /external/tensorflow/ |
| D | RELEASE.md | 559 * Add `num_parallel_reads` argument to `tf.data.TFRecordDataset`.
|