Searched refs:prefetch_buffer_size (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | readers.py | 234 prefetch_buffer_size=None, argument
291 if prefetch_buffer_size is None:
292 prefetch_buffer_size = dataset_ops.AUTOTUNE
319 if prefetch_buffer_size == 0:
322 return dataset.prefetch(buffer_size=prefetch_buffer_size)
341 prefetch_buffer_size=None, argument
482 if prefetch_buffer_size is None:
483 prefetch_buffer_size = dataset_ops.AUTOTUNE
624 dataset = dataset.prefetch(prefetch_buffer_size)
645 prefetch_buffer_size=None, argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | multi_device_iterator_ops.py | 207 def _create_device_dataset(prototype_ds, incarnation_id, prefetch_buffer_size, argument
211 if prefetch_buffer_size > 0:
213 ds = dataset_ops.PrefetchDataset(ds, prefetch_buffer_size, slack_period=1)
215 ds = ds.prefetch(prefetch_buffer_size)
226 prefetch_buffer_size=1, argument
249 self._prefetch_buffer_size = prefetch_buffer_size
430 prefetch_buffer_size=1, argument
490 if prefetch_buffer_size > max_buffer_size:
491 max_buffer_size = prefetch_buffer_size
529 incarnation_id, prefetch_buffer_size,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | multi_device_iterator_test.py | 74 dataset, [self._devices[1], self._devices[2]], prefetch_buffer_size=0)
149 max_buffer_size=[0, 1, 10], prefetch_buffer_size=[0, 1, 10])))
150 def testBasic(self, prefetch_buffer_size, max_buffer_size): argument
155 prefetch_buffer_size=prefetch_buffer_size)
274 dataset, [self._devices[1], self._devices[2]], prefetch_buffer_size=4)
289 dataset, [self._devices[1], self._devices[2]], prefetch_buffer_size=4)
330 max_buffer_size=[0, 1, 10], prefetch_buffer_size=[0, 1, 10])))
331 def testBasic(self, max_buffer_size, prefetch_buffer_size): argument
337 prefetch_buffer_size=prefetch_buffer_size)
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.data.experimental.pbtxt | 165 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'reader…
169 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_pa…
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.data.experimental.pbtxt | 189 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'reader…
193 …ochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_pa…
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | input_lib.py | 335 prefetch_buffer_size=self._options
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | input_lib.py | 1780 prefetch_buffer_size=self._options
|