Searched defs:arg_dimensions (Results 1 – 1 of 1) sorted by relevance
3672 absl::Span<const int64_t> arg_dimensions = arg_shape.dimensions(); in GenerateReduceOutputElement() local3747 absl::Span<const int64_t> arg_dimensions = arg_shape.dimensions(); in HandleReduce() local