Skip to content

MLBEDSW-11298: Fix cascading of TOSA AveragePool

Alexander Bengtsson requested to merge MLBEDSW-11298 into main
  • AveragePool requires QuantizePoolingScale to normalise the kernel-sum after pooling. Before this patch, RescalePooling was called for every stripe of a cascade. This would re-calculate the OFM-scaling every stripe.
  • Fix by only calling RescalePooling for TFLITE quantization and calling QuantizePoolingScale explicitly in RCS for the TOSA case.

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

Merge request reports

Loading