Searched defs:prefetch_buffer_size (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | readers.py | 234 prefetch_buffer_size=None, argument
341 prefetch_buffer_size=None, argument
645 prefetch_buffer_size=None, argument
926 prefetch_buffer_size=None, argument
1110 prefetch_buffer_size=None, argument
|
| /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
226 prefetch_buffer_size=1, argument
430 prefetch_buffer_size=1, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | multi_device_iterator_test.py | 150 def testBasic(self, prefetch_buffer_size, max_buffer_size): argument
331 def testBasic(self, max_buffer_size, prefetch_buffer_size): argument
|