Searched defs:buffer_output_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 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 | 173 int64_t buffer_output_elements, in ComputeMaxBufferedElements() 200 int64_t block_length, int64_t buffer_output_elements, in Dataset() 1713 int64_t buffer_output_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 | 33 buffer_output_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() 1258 int64_t buffer_output_elements = 0; in MakeDataset() local
|
H A D | parallel_interleave_dataset_op_test.cc | 31 const std::string& deterministic, int64_t buffer_output_elements, in ParallelInterleaveDatasetParams()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
H A D | readers.py | 326 buffer_output_elements, argument
|
H A D | dataset_ops.py | 5569 buffer_output_elements=AUTOTUNE, argument
|