Searched refs:result_to_arg_index (Results 1 – 1 of 1) sorted by relevance
2362 absl::Span<const int64> result_to_arg_index) { in GenerateReduceOutputElement() argument2370 base[result_to_arg_index[i]] = output_index[i]; in GenerateReduceOutputElement()2460 std::vector<int64> result_to_arg_index; in HandleReduce() local2463 result_to_arg_index.push_back(i); in HandleReduce()2478 arg_dim_steps, arg_dim_counts, result_to_arg_index); in HandleReduce()