Searched refs:next_input_element (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_reducer_dataset_op.cc | 212 std::vector<Tensor> next_input_element; in GetNextInternal() local
214 input_impl_->GetNext(ctx, &next_input_element, &end_of_input_)); in GetNextInternal()
220 ctx, next_input_element, &key_func_output)); in GetNextInternal()
241 args.reserve(states_[key].size() + next_input_element.size()); in GetNextInternal()
244 std::copy(next_input_element.begin(), next_input_element.end(), in GetNextInternal()
|
| D | group_by_window_dataset_op.cc | 213 std::vector<Tensor> next_input_element; in GetNextInternal() local
215 input_impl_->GetNext(ctx, &next_input_element, &end_of_input_)); in GetNextInternal()
222 ctx, next_input_element, &key_func_output)); in GetNextInternal()
260 group.push_back(std::move(next_input_element)); in GetNextInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 700 std::vector<Tensor> next_input_element; in ComputeAsync() local
702 status = raw_iterator->GetNext(&iter_ctx, &next_input_element, in ComputeAsync()
710 args.reserve(state.size() + next_input_element.size()); in ComputeAsync()
712 std::copy(next_input_element.begin(), next_input_element.end(), in ComputeAsync()
|