Searched refs:reduce_func_output (Results 1 – 2 of 2) sorted by relevance
247 std::vector<Tensor> reduce_func_output; in GetNextInternal() local249 ctx, std::move(args), &reduce_func_output)); in GetNextInternal()250 states_[key] = reduce_func_output; in GetNextInternal()
715 std::vector<Tensor> reduce_func_output; in ComputeAsync() local717 &reduce_func_output); in ComputeAsync()721 std::swap(reduce_func_output, state); in ComputeAsync()