xla

88b80f8f - Make sure the real sizes are accounted when averaging reduces, to deal with dynamic shapes.