Home
last modified time | relevance | path

Searched refs:prefetch_input_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))
174 sloppy=[False, True], prefetch_input_elements=[0, 1])))
175 def testSingleThreaded(self, sloppy, prefetch_input_elements): argument
187 prefetch_input_elements=prefetch_input_elements))
208 prefetch_input_elements=1))
240 prefetch_input_elements=1))
280 prefetch_input_elements=1))
318 prefetch_input_elements=1))
359 prefetch_input_elements=1))
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DParallelInterleaveDataset.pbtxt28 name: "prefetch_input_elements"
84 name: "prefetch_input_elements"
147 name: "prefetch_input_elements"
217 name: "prefetch_input_elements"
H A DLegacyParallelInterleaveDatasetV2.pbtxt24 name: "prefetch_input_elements"
83 name: "prefetch_input_elements"
149 name: "prefetch_input_elements"
222 name: "prefetch_input_elements"
H A DParallelInterleaveDatasetV4.pbtxt24 name: "prefetch_input_elements"
87 name: "prefetch_input_elements"
157 name: "prefetch_input_elements"
234 name: "prefetch_input_elements"
H A DExperimentalParallelInterleaveDataset.pbtxt28 name: "prefetch_input_elements"
84 name: "prefetch_input_elements"
147 name: "prefetch_input_elements"
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dparallel_interleave_dataset_op.cc172 int64_t ComputeMaxBufferedElements(int64_t prefetch_input_elements, in ComputeMaxBufferedElements() argument
175 return (prefetch_input_elements + cycle_length) * buffer_output_elements; in ComputeMaxBufferedElements()
201 int64_t prefetch_input_elements, int64_t num_parallel_calls, in Dataset() argument
213 prefetch_input_elements, cycle_length_)), in Dataset()
1714 int64_t prefetch_input_elements = model::kAutotune; in MakeDataset() local
1726 &prefetch_input_elements)); in MakeDataset()
1728 prefetch_input_elements == model::kAutotune || in MakeDataset()
1729 prefetch_input_elements >= 0, in MakeDataset()
1732 prefetch_input_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
46 prefetch_input_elements_(prefetch_input_elements), in ParallelInterleaveDatasetParams()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dinterleave_ops.py34 prefetch_input_elements=None): argument
84 prefetch_input_elements)
H A Dreaders.py607 prefetch_input_elements=None)
1062 prefetch_input_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
110 prefetch_input_elements_(prefetch_input_elements), in Dataset()
1264 int64_t prefetch_input_elements = 0; in MakeDataset() local
1266 &prefetch_input_elements)); in MakeDataset()
1268 ctx, prefetch_input_elements >= 0, in MakeDataset()
1278 prefetch_input_elements, output_types_, output_shapes_, in MakeDataset()
H A Dparallel_interleave_dataset_op_test.cc32 int64_t prefetch_input_elements, FunctionDefHelper::AttrValueWrapper func, in ParallelInterleaveDatasetParams() argument
43 prefetch_input_elements_(prefetch_input_elements), in ParallelInterleaveDatasetParams()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Dreaders.py111 prefetch_input_elements=None,
327 prefetch_input_elements, argument
348 prefetch_input_elements,
H A Ddataset_ops.py5570 prefetch_input_elements=AUTOTUNE, argument
5591 prefetch_input_elements,
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DParallelInterleaveDataset.pbtxt28 name: "prefetch_input_elements"
H A DExperimentalParallelInterleaveDataset.pbtxt28 name: "prefetch_input_elements"
H A DLegacyParallelInterleaveDatasetV2.pbtxt24 name: "prefetch_input_elements"
H A DParallelInterleaveDatasetV4.pbtxt24 name: "prefetch_input_elements"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_ParallelInterleaveDataset.pbtxt50 name: "prefetch_input_elements"
H A Dapi_def_ParallelInterleaveDatasetV4.pbtxt40 name: "prefetch_input_elements"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.data.experimental.pbtxt181 …h\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefetch_input_elements\'], varar…
H A Dtensorflow.raw_ops.pbtxt1561 …h\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefetch_input_elements\', \'f\',…
2205 …'cycle_length\', \'block_length\', \'buffer_output_elements\', \'prefetch_input_elements\', \'f\',…
2885 …h\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefetch_input_elements\', \'f\',…
2897 …'cycle_length\', \'block_length\', \'buffer_output_elements\', \'prefetch_input_elements\', \'num_…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.data.experimental.pbtxt209 …h\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefetch_input_elements\'], varar…
H A Dtensorflow.raw_ops.pbtxt1561 …h\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefetch_input_elements\', \'f\',…
2205 …'cycle_length\', \'block_length\', \'buffer_output_elements\', \'prefetch_input_elements\', \'f\',…
2885 …h\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefetch_input_elements\', \'f\',…
2897 …'cycle_length\', \'block_length\', \'buffer_output_elements\', \'prefetch_input_elements\', \'num_…
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt17277 name: "prefetch_input_elements"
23832 name: "prefetch_input_elements"
30740 name: "prefetch_input_elements"
30983 name: "prefetch_input_elements"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/
H A Dops_history.v2.pbtxt23898 name: "prefetch_input_elements"