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 | 224 std::vector<Tensor> next_input_element; in GetNextInternal() local
226 input_impl_->GetNext(ctx, &next_input_element, &end_of_input_)); in GetNextInternal()
232 ctx, next_input_element, &key_func_output, model_node())); in GetNextInternal()
254 args.reserve(states_[key].size() + next_input_element.size()); in GetNextInternal()
257 std::copy(next_input_element.begin(), next_input_element.end(), in GetNextInternal()
|
| D | group_by_window_dataset_op.cc | 233 std::vector<Tensor> next_input_element; in GetNextInternal() local
235 input_impl_->GetNext(ctx, &next_input_element, &end_of_input_)); in GetNextInternal()
242 ctx, next_input_element, &key_func_output, model_node())); in GetNextInternal()
281 group.push_back(std::move(next_input_element)); in GetNextInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 729 std::vector<Tensor> next_input_element; in DoCompute() local
732 iterator->GetNext(&iter_ctx, &next_input_element, &end_of_input)); in DoCompute()
739 args.reserve(state.size() + next_input_element.size()); in DoCompute()
741 std::copy(next_input_element.begin(), next_input_element.end(), in DoCompute()
|