Searched refs:arg_dim_counts (Results 1 – 1 of 1) sorted by relevance
1464 std::vector<int64> arg_dim_counts(arg_dimensions.size()); in HandleReduce() local1467 arg_dim_counts[dim] = arg_dimensions[dim]; in HandleReduce()1514 ShapeUtil::ForEachIndex(arg_literal.shape(), base, arg_dim_counts, in HandleReduce()