Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dbatch_dataset_op.cc162 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
Dpadded_batch_dataset_op.cc197 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
Dparallel_batch_dataset_op.cc332 auto batch_elements = in CallBatching() local
Ddataset_utils.cc849 std::vector<std::vector<Tensor>>* batch_elements) { in CopyBatch()
/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