Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
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()
Dbatch_dataset_op.cc162 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
Dparallel_batch_dataset_op.cc367 auto batch_elements = in CallBatching() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc169 std::vector<Tensor> batch_elements; in GetNextInternal() local
Dsliding_window_dataset_op.cc156 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
/external/tensorflow/tensorflow/core/data/
Ddataset_utils.cc722 const std::vector<std::vector<Tensor>>& batch_elements, in CopyBatch()