Searched refs:batch_element_tuple (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 188 std::vector<Tensor> batch_element_tuple; in GetNextInternal() local
189 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &batch_element_tuple, in GetNextInternal()
192 DCHECK_EQ(1, batch_element_tuple.size()); in GetNextInternal()
193 batch_elements.push_back(std::move(batch_element_tuple[0])); in GetNextInternal()
194 total_elements += batch_element_tuple[0].NumElements(); in GetNextInternal()
198 if (batch_element_tuple[0].shape().dims() != row_ndims) { in GetNextInternal()
201 batch_element_tuple[0].shape().DebugString(), in GetNextInternal()
209 std::max(batch_element_tuple[0].dim_size(j), in GetNextInternal()
211 } else if (batch_element_tuple[0].dim_size(j) > in GetNextInternal()
215 batch_element_tuple[0].shape().DebugString(), in GetNextInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 148 std::vector<Tensor> batch_element_tuple; in GetNextInternal() local
149 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &batch_element_tuple, in GetNextInternal()
152 batch_elements.emplace_back(std::move(batch_element_tuple)); in GetNextInternal()
|
| D | padded_batch_dataset_op.cc | 238 std::vector<Tensor> batch_element_tuple; in GetNextInternal() local
239 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &batch_element_tuple, in GetNextInternal()
242 batch_elements.push_back(std::move(batch_element_tuple)); in GetNextInternal()
|