Searched defs:prefetch_input_elements (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | parallel_interleave_test.py | 58 buffer_output_elements, prefetch_input_elements): argument
175 def testSingleThreaded(self, sloppy, prefetch_input_elements): argument
420 def testPartiallyEmptyOutputs(self, sloppy, prefetch_input_elements): argument
634 buffer_output_elements, prefetch_input_elements): argument
678 buffer_output_elements, prefetch_input_elements): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | parallel_interleave_dataset_op.cc | 172 int64_t ComputeMaxBufferedElements(int64_t prefetch_input_elements, in ComputeMaxBufferedElements()
201 int64_t prefetch_input_elements, int64_t num_parallel_calls, in Dataset()
1714 int64_t prefetch_input_elements = model::kAutotune; in MakeDataset() local
|
| H A D | parallel_interleave_dataset_op_test.cc | 34 int64_t buffer_output_elements, int64_t prefetch_input_elements, in ParallelInterleaveDatasetParams()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | interleave_ops.py | 34 prefetch_input_elements=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | parallel_interleave_dataset_op.cc | 100 int64_t buffer_output_elements, int64_t prefetch_input_elements, in Dataset()
1264 int64_t prefetch_input_elements = 0; in MakeDataset() local
|
| H A D | parallel_interleave_dataset_op_test.cc | 32 int64_t prefetch_input_elements, FunctionDefHelper::AttrValueWrapper func, in ParallelInterleaveDatasetParams()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | readers.py | 327 prefetch_input_elements, argument
|
| H A D | dataset_ops.py | 5570 prefetch_input_elements=AUTOTUNE, argument
|