Searched defs:arg_dim_counts (Results 1 – 1 of 1) sorted by relevance
2361 absl::Span<const int64> arg_dim_counts, in GenerateReduceOutputElement()2448 std::vector<int64> arg_dim_counts(arg_dimensions.size()); in HandleReduce() local