Searched refs:batch_element_tuple (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | dense_to_sparse_batch_dataset_op.cc | 196 std::vector<Tensor> batch_element_tuple; in GetNextInternal() local
197 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &batch_element_tuple, in GetNextInternal()
200 DCHECK_EQ(1, batch_element_tuple.size()); in GetNextInternal()
201 batch_elements.push_back(std::move(batch_element_tuple[0])); in GetNextInternal()
202 total_elements += batch_element_tuple[0].NumElements(); in GetNextInternal()
206 if (batch_element_tuple[0].shape().dims() != row_ndims) { in GetNextInternal()
209 batch_element_tuple[0].shape().DebugString(), in GetNextInternal()
217 std::max(batch_element_tuple[0].dim_size(j), in GetNextInternal()
219 } else if (batch_element_tuple[0].dim_size(j) > in GetNextInternal()
223 batch_element_tuple[0].shape().DebugString(), in GetNextInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | batch_dataset_op.cc | 149 std::vector<Tensor> batch_element_tuple; in Get() local
150 TF_RETURN_IF_ERROR(input_->Get(ctx, i, &batch_element_tuple)); in Get()
151 batch_elements.emplace_back(std::move(batch_element_tuple)); in Get()
202 std::vector<Tensor> batch_element_tuple; in GetNextInternal() local
204 input_impl_->GetNext(ctx, &batch_element_tuple, end_of_sequence)); in GetNextInternal()
206 batch_elements.emplace_back(std::move(batch_element_tuple)); in GetNextInternal()
|
| H A D | padded_batch_dataset_op.cc | 210 std::vector<Tensor> batch_element_tuple; in GetNextInternal() local
211 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &batch_element_tuple, in GetNextInternal()
214 batch_elements.push_back(std::move(batch_element_tuple)); in GetNextInternal()
|
| H A D | parallel_batch_dataset_op.cc | 378 std::vector<Tensor> batch_element_tuple; in CallBatching() local
379 Status status = input_impl_->GetNext(ctx.get(), &batch_element_tuple, in CallBatching()
388 batch_elements->emplace_back(std::move(batch_element_tuple)); in CallBatching()
|