Searched refs:prefetch_buffer_size (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 230 prefetch_buffer_size=None, argument
287 if prefetch_buffer_size is None:
288 prefetch_buffer_size = dataset_ops.AUTOTUNE
315 if prefetch_buffer_size == 0:
318 return dataset.prefetch(buffer_size=prefetch_buffer_size)
337 prefetch_buffer_size=None, argument
476 if prefetch_buffer_size is None:
477 prefetch_buffer_size = dataset_ops.AUTOTUNE
605 dataset = dataset.prefetch(prefetch_buffer_size)
626 prefetch_buffer_size=None, argument
[all …]
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 193 def _create_device_dataset(prototype_ds, incarnation_id, prefetch_buffer_size, argument
197 if prefetch_buffer_size > 0:
199 ds = dataset_ops.PrefetchDataset(ds, prefetch_buffer_size, slack_period=1)
201 ds = ds.prefetch(prefetch_buffer_size)
218 prefetch_buffer_size=1, argument
241 self._prefetch_buffer_size = prefetch_buffer_size
474 prefetch_buffer_size=1, argument
530 if prefetch_buffer_size > max_buffer_size:
531 max_buffer_size = prefetch_buffer_size
564 incarnation_id, prefetch_buffer_size,
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | multi_device_iterator_test.py | 64 max_buffer_size=[0, 1, 10], prefetch_buffer_size=[0, 1, 10])))
65 def testBasic(self, prefetch_buffer_size, max_buffer_size): argument
70 prefetch_buffer_size=prefetch_buffer_size)
216 dataset, ["/cpu:1", "/cpu:2"], prefetch_buffer_size=4)
241 dataset, ["/cpu:1", "/cpu:2"], prefetch_buffer_size=4)
360 max_buffer_size=[0, 1, 10], prefetch_buffer_size=[0, 1, 10])))
361 def testBasic(self, max_buffer_size, prefetch_buffer_size): argument
371 prefetch_buffer_size=prefetch_buffer_size)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 157 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'reader…
161 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_pa…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 181 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'reader…
185 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_pa…
|