Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_utils.cc744 const size_t num_tuple_components = batch_elements.at(0).size(); in CopyBatch() local
745 out_tensors->reserve(num_tuple_components); in CopyBatch()
747 for (size_t component_index = 0; component_index < num_tuple_components; in CopyBatch()
764 for (size_t component_index = 0; component_index < num_tuple_components; in CopyBatch()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dsliding_window_dataset_op.cc222 const size_t num_tuple_components = batch_elements[0].size(); in GetNextInternal() local
224 for (size_t component_index = 0; component_index < num_tuple_components; in GetNextInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dwindow_dataset_op.cc232 const size_t num_tuple_components = window_elements[0].size(); in GetNextInternal() local
235 for (size_t idx = 0; idx < num_tuple_components; ++idx) { in GetNextInternal()
H A Dpadded_batch_dataset_op.cc283 const size_t num_tuple_components = batch_elements[0].size(); in CopyBatch() local
285 for (size_t component_index = 0; component_index < num_tuple_components; in CopyBatch()