Home
last modified time | relevance | path

Searched refs:buffer_output_elements_ (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dparallel_interleave_dataset_op_test.cc42 buffer_output_elements_(buffer_output_elements), in ParallelInterleaveDatasetParams()
63 CreateTensor<int64_t>(TensorShape({}), {buffer_output_elements_})); in GetInputTensors()
105 int64_t buffer_output_elements_; member in tensorflow::data::experimental::__anon0cf95fbd0111::ParallelInterleaveDatasetParams
H A Dparallel_interleave_dataset_op.cc109 buffer_output_elements_(buffer_output_elements), in Dataset()
197 b->AddScalar(buffer_output_elements_, &buffer_output_elements_node)); in AsGraphDefInternal()
837 dataset()->buffer_output_elements_) { in WorkerThread()
898 dataset()->buffer_output_elements_) { in WorkerThread()
1209 const int64_t buffer_output_elements_; member in tensorflow::data::experimental::ParallelInterleaveDatasetOp::Dataset
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dparallel_interleave_dataset_op_test.cc45 buffer_output_elements_(buffer_output_elements), in ParallelInterleaveDatasetParams()
68 CreateTensor<int64_t>(TensorShape({}), {buffer_output_elements_})); in GetInputTensors()
112 int64_t buffer_output_elements_; member in tensorflow::data::__anonf2f0f7400111::ParallelInterleaveDatasetParams
H A Dparallel_interleave_dataset_op.cc210 buffer_output_elements_( in Dataset()
311 b->AddScalar(buffer_output_elements_, &buffer_output_elements_node)); in AsGraphDefInternal()
467 dataset()->buffer_output_elements_, in CreateNode()
1060 if (element->results.size() == dataset()->buffer_output_elements_) { in ProcessElement()
1162 element->results.size() < dataset()->buffer_output_elements_; in NeedsProcessing()
1669 const int64_t buffer_output_elements_; member in tensorflow::data::ParallelInterleaveDatasetOp::Dataset