Skip to content

MLBEDSW-9408: Add full REDUCE MIN/MAX/SUM decompostion

Johan Gunnarsson requested to merge MLBEDSW-9408 into main

This implements full decomposition of TOSA REDUCE MIN/MAX/ANY/ALL of the reduced axis.

  • Extend decomposition to do blockwise reduce operations in the reduced axis.
  • Add ReduceSum/ReduceMinMax to constraints.
  • Move reshaping of reduce ops into decomposition.
  • Move creating a reduce ops kernel to ConvertAttributes.
  • Remove RewriteReduceMinMaxAnyAll.

Signed-off-by: Johan Gunnarsson johan.gunnarsson@arm.com Change-Id: Ic0873dab1c3c5344045590d1c11724986b896120

Merge request reports