Searched refs:batch_element_tuple (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| 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()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 172 std::vector<Tensor> batch_element_tuple; in GetNextInternal() local
174 input_impl_->GetNext(ctx, &batch_element_tuple, end_of_sequence)); in GetNextInternal()
176 batch_elements.emplace_back(std::move(batch_element_tuple)); in GetNextInternal()
|
| D | padded_batch_dataset_op.cc | 208 std::vector<Tensor> batch_element_tuple; in GetNextInternal() local
209 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &batch_element_tuple, in GetNextInternal()
212 batch_elements.push_back(std::move(batch_element_tuple)); in GetNextInternal()
|
| D | parallel_batch_dataset_op.cc | 338 std::vector<Tensor> batch_element_tuple; in CallBatching() local
339 Status status = input_impl_->GetNext(ctx.get(), &batch_element_tuple, in CallBatching()
348 batch_elements->emplace_back(std::move(batch_element_tuple)); in CallBatching()
|