Searched defs:arg_dimensions (Results 1 – 1 of 1) sorted by relevance
2431 absl::Span<const int64> arg_dimensions = AsInt64Slice(arg_shape.dimensions()); in GenerateReduceOutputElement() local2506 absl::Span<const int64> arg_dimensions = AsInt64Slice(arg_shape.dimensions()); in HandleReduce() local