Home
last modified time | relevance | path

Searched refs:buffer_output_elements (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A Dparallel_interleave_test.py58 buffer_output_elements, prefetch_input_elements): argument
82 buffer_output_elements, prefetch_input_elements))
186 buffer_output_elements=1,
207 buffer_output_elements=1,
239 buffer_output_elements=1,
279 buffer_output_elements=1,
317 buffer_output_elements=1,
358 buffer_output_elements=1,
393 buffer_output_elements=1,
410 buffer_output_elements=1,
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DParallelInterleaveDataset.pbtxt24 name: "buffer_output_elements"
80 name: "buffer_output_elements"
143 name: "buffer_output_elements"
213 name: "buffer_output_elements"
H A DLegacyParallelInterleaveDatasetV2.pbtxt20 name: "buffer_output_elements"
79 name: "buffer_output_elements"
145 name: "buffer_output_elements"
218 name: "buffer_output_elements"
H A DParallelInterleaveDatasetV4.pbtxt20 name: "buffer_output_elements"
83 name: "buffer_output_elements"
153 name: "buffer_output_elements"
230 name: "buffer_output_elements"
H A DExperimentalParallelInterleaveDataset.pbtxt24 name: "buffer_output_elements"
80 name: "buffer_output_elements"
143 name: "buffer_output_elements"
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dparallel_interleave_dataset_op.cc173 int64_t buffer_output_elements, in ComputeMaxBufferedElements() argument
175 return (prefetch_input_elements + cycle_length) * buffer_output_elements; in ComputeMaxBufferedElements()
200 int64_t block_length, int64_t buffer_output_elements, in Dataset() argument
211 ComputeBufferOutputElements(buffer_output_elements, block_length)), in Dataset()
1713 int64_t buffer_output_elements = model::kAutotune; in MakeDataset() local
1717 &buffer_output_elements)); in MakeDataset()
1719 buffer_output_elements == model::kAutotune || in MakeDataset()
1720 buffer_output_elements > 0, in MakeDataset()
1723 buffer_output_elements)); in MakeDataset()
1768 buffer_output_elements, prefetch_input_elements, num_parallel_calls, in MakeDataset()
H A Dparallel_interleave_dataset_op_test.cc34 int64_t buffer_output_elements, int64_t prefetch_input_elements, in ParallelInterleaveDatasetParams() argument
45 buffer_output_elements_(buffer_output_elements), in ParallelInterleaveDatasetParams()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dinterleave_ops.py33 buffer_output_elements=None, argument
83 buffer_output_elements,
H A Dreaders.py606 buffer_output_elements=None,
1061 buffer_output_elements=None,
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dparallel_interleave_dataset_op.cc100 int64_t buffer_output_elements, int64_t prefetch_input_elements, in Dataset() argument
109 buffer_output_elements_(buffer_output_elements), in Dataset()
1258 int64_t buffer_output_elements = 0; in MakeDataset() local
1260 &buffer_output_elements)); in MakeDataset()
1261 OP_REQUIRES(ctx, buffer_output_elements > 0, in MakeDataset()
1277 block_length, deterministic_, buffer_output_elements, in MakeDataset()
H A Dparallel_interleave_dataset_op_test.cc31 const std::string& deterministic, int64_t buffer_output_elements, in ParallelInterleaveDatasetParams() argument
42 buffer_output_elements_(buffer_output_elements), in ParallelInterleaveDatasetParams()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Dreaders.py110 buffer_output_elements=None,
326 buffer_output_elements, argument
344 buffer_output_elements,
H A Ddataset_ops.py5569 buffer_output_elements=AUTOTUNE, argument
5587 buffer_output_elements,
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DParallelInterleaveDataset.pbtxt24 name: "buffer_output_elements"
H A DExperimentalParallelInterleaveDataset.pbtxt24 name: "buffer_output_elements"
H A DLegacyParallelInterleaveDatasetV2.pbtxt20 name: "buffer_output_elements"
H A DParallelInterleaveDatasetV4.pbtxt20 name: "buffer_output_elements"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_ParallelInterleaveDataset.pbtxt43 name: "buffer_output_elements"
H A Dapi_def_ParallelInterleaveDatasetV4.pbtxt33 name: "buffer_output_elements"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.data.experimental.pbtxt181 …[\'map_func\', \'cycle_length\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefe…
H A Dtensorflow.raw_ops.pbtxt1561 …r_arguments\', \'cycle_length\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefe…
2205 …et\', \'other_arguments\', \'cycle_length\', \'block_length\', \'buffer_output_elements\', \'prefe…
2885 …r_arguments\', \'cycle_length\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefe…
2897 …et\', \'other_arguments\', \'cycle_length\', \'block_length\', \'buffer_output_elements\', \'prefe…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.data.experimental.pbtxt209 …[\'map_func\', \'cycle_length\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefe…
H A Dtensorflow.raw_ops.pbtxt1561 …r_arguments\', \'cycle_length\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefe…
2205 …et\', \'other_arguments\', \'cycle_length\', \'block_length\', \'buffer_output_elements\', \'prefe…
2885 …r_arguments\', \'cycle_length\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefe…
2897 …et\', \'other_arguments\', \'cycle_length\', \'block_length\', \'buffer_output_elements\', \'prefe…
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt17273 name: "buffer_output_elements"
23828 name: "buffer_output_elements"
30736 name: "buffer_output_elements"
30979 name: "buffer_output_elements"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/
H A Dops_history.v2.pbtxt23894 name: "buffer_output_elements"