| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | make_tf_record_dataset_test.py | 38 num_parallel_reads=1, drop_final_batch=False, parser_fn=False): argument
55 num_parallel_reads=num_parallel_reads,
63 interleave_cycle_length=num_parallel_reads,
83 self._read_test(batch_size, num_epochs, num_parallel_reads=8)
96 self._read_test(batch_size, num_epochs, num_parallel_reads=8,
106 self._read_test(batch_size, num_epochs, num_parallel_reads=8,
109 def _shuffle_test(self, batch_size, num_epochs, num_parallel_reads=1, argument
117 num_parallel_reads=num_parallel_reads,
158 for num_parallel_reads in [1, 2]:
160 self._shuffle_test(batch_size, num_epochs, num_parallel_reads)
[all …]
|
| D | auto_shard_dataset_test.py | 257 num_parallel_reads=parallel_reads,
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | readers.py | 75 def _create_dataset_reader(dataset_creator, filenames, num_parallel_reads=None): argument
92 if num_parallel_reads is None:
94 elif num_parallel_reads == dataset_ops.AUTOTUNE:
96 read_one_file, num_parallel_calls=num_parallel_reads)
101 cycle_length=num_parallel_reads,
150 num_parallel_reads=None): argument
181 num_parallel_reads)
199 num_parallel_reads=None): argument
201 num_parallel_reads)
310 num_parallel_reads=None): argument
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 231 num_parallel_reads=None, argument
275 if num_parallel_reads is None:
280 num_parallel_reads = 24
294 files, num_parallel_reads=num_parallel_reads)
338 num_parallel_reads=None, argument
473 if num_parallel_reads is None:
474 num_parallel_reads = 1
572 if num_parallel_reads == dataset_ops.AUTOTUNE:
574 filename_to_dataset, num_parallel_calls=num_parallel_reads)
584 cycle_length=num_parallel_reads,
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | datasets.py | 60 num_parallel_reads: Optional[int] = None,
128 num_parallel_reads = num_parallel_reads or 8
152 reader_fn, cycle_length=num_parallel_reads, sloppy=sloppy))
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | fixed_length_record_dataset_test.py | 168 num_parallel_reads=4)
204 num_parallel_reads=4)
|
| D | text_line_dataset_test.py | 130 dataset = readers.TextLineDataset(files, num_parallel_reads=4)
|
| D | tf_record_dataset_test.py | 187 dataset = readers.TFRecordDataset(files, num_parallel_reads=4)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.-t-f-record-dataset.pbtxt | 12 …[\'self\', \'filenames\', \'compression_type\', \'buffer_size\', \'num_parallel_reads\'], varargs=…
|
| D | tensorflow.data.-text-line-dataset.pbtxt | 13 …[\'self\', \'filenames\', \'compression_type\', \'buffer_size\', \'num_parallel_reads\'], varargs=…
|
| D | tensorflow.data.-fixed-length-record-dataset.pbtxt | 13 …ytes\', \'footer_bytes\', \'buffer_size\', \'compression_type\', \'num_parallel_reads\'], varargs=…
|
| D | tensorflow.data.experimental.pbtxt | 161 …uffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_parallel_reads\', \'sloppy\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.-t-f-record-dataset.pbtxt | 26 …[\'self\', \'filenames\', \'compression_type\', \'buffer_size\', \'num_parallel_reads\'], varargs=…
|
| D | tensorflow.data.-text-line-dataset.pbtxt | 26 …[\'self\', \'filenames\', \'compression_type\', \'buffer_size\', \'num_parallel_reads\'], varargs=…
|
| D | tensorflow.data.-fixed-length-record-dataset.pbtxt | 26 …ytes\', \'footer_bytes\', \'buffer_size\', \'compression_type\', \'num_parallel_reads\'], varargs=…
|
| D | tensorflow.data.experimental.pbtxt | 185 …uffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_parallel_reads\', \'sloppy\…
|
| /external/tensorflow/ |
| D | RELEASE.md | 1850 * Add `num_parallel_reads` and passing in a Dataset containing filenames
2489 * Add num_parallel_reads and passing in a Dataset containing filenames
3232 * Add `num_parallel_reads` argument to `tf.data.TFRecordDataset`.
|