Searched refs:reduce_func_output (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_reducer_dataset_op.cc | 260 std::vector<Tensor> reduce_func_output; in GetNextInternal() local
262 ctx, std::move(args), &reduce_func_output, model_node())); in GetNextInternal()
263 states_[key] = reduce_func_output; in GetNextInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 744 std::vector<Tensor> reduce_func_output; in DoCompute() local
746 &iter_ctx, std::move(args), &reduce_func_output, /*node=*/nullptr)); in DoCompute()
747 if (reduce_func_output.size() != state.size()) { in DoCompute()
752 state.size(), ", output=", reduce_func_output.size(), ")."); in DoCompute()
754 std::swap(reduce_func_output, state); in DoCompute()
|