Skip to content

MLBEDSW-10333: Fixes for RewriteReduceSum

Alexander Bengtsson requested to merge MLBEDSW-10333 into main
  • Maintain rounding when lowering ReduceSum.
  • Fix shape of intermediate tensor when ReduceSum is lowered to ReduceSum + Sub (shape should be inherited from the tensor connection)
  • Keep quantization on the ReduceSum when ReduceSum is lowered to Transpose + ReduceSum.

Change-Id: I9653f3638bbda8fa287a3f8a32c4bd3abe4e79ae Signed-off-by: Alexander Bengtsson Alexander.Bengtsson@arm.com

Merge request reports