Searched refs:prefetch_buffer_size (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 227 prefetch_buffer_size=optimization.AUTOTUNE, argument
305 if prefetch_buffer_size == 0:
308 return dataset.prefetch(buffer_size=prefetch_buffer_size)
327 prefetch_buffer_size=optimization.AUTOTUNE, argument
514 dataset = dataset.prefetch(prefetch_buffer_size)
535 prefetch_buffer_size=optimization.AUTOTUNE, argument
546 prefetch_buffer_size, num_parallel_reads, sloppy, num_rows_for_inference,
719 prefetch_buffer_size=optimization.AUTOTUNE, argument
866 dataset = dataset.prefetch(prefetch_buffer_size)
881 prefetch_buffer_size=optimization.AUTOTUNE, argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | readers.py | 48 prefetch_buffer_size=optimization.AUTOTUNE, argument
141 prefetch_buffer_size, num_parallel_reads, sloppy, num_rows_for_inference,
176 prefetch_buffer_size=optimization.AUTOTUNE, argument
273 prefetch_buffer_size, reader_num_threads, parser_num_threads,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 141 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'reader…
145 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_pa…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 141 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'reader…
145 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_pa…
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 197 prefetch_buffer_size=1, argument
221 self._prefetch_buffer_size = prefetch_buffer_size
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | multi_device_iterator_test.py | 203 dataset, ["/cpu:1", "/cpu:2"], prefetch_buffer_size=4)
226 dataset, ["/cpu:1", "/cpu:2"], prefetch_buffer_size=4)
|