Home
last modified time | relevance | path

Searched defs:batch_elements (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dbatch_dataset_op.cc145 std::vector<std::vector<Tensor>> batch_elements; in Get() local
192 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
H A Dpadded_batch_dataset_op.cc199 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
281 const std::vector<std::vector<Tensor>>& batch_elements, in CopyBatch()
H A Dparallel_batch_dataset_op.cc372 auto batch_elements = in CallBatching() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Ddense_to_sparse_batch_dataset_op.cc169 std::vector<Tensor> batch_elements; in GetNextInternal() local
H A Dsliding_window_dataset_op.cc172 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_utils.cc740 const std::vector<std::vector<Tensor>>& batch_elements, in CopyBatch()