commit 5fc0fd6661f9647092deb052d052973a237bd52d
author Matthew Sloyan <matthew.sloyan@arm.com> Mon May 03 12:22:03 2021 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Thu May 20 13:57:36 2021 +0000
tree f46c8ad169414ae400b22b046e25cedf7ba76c9d
parent 1bed2f4f297d16f738b2c14f9e0fd24b7b03dade

MLCE-418 Reduce layer does not support multiple axes

 * Added backend specific optimization to chain new reduces layers
   for each axis to simulate behaviour of a layer with multiple axes.
 * Added function to calculate reduced output shape.
 * Added unit tests.
 * Includes rework to fix IVGCVSW-5987.

Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I154b3698b5e6756b05b2a0b5a3f0896184efce72

Android.mk

diff --git a/Android.mk b/Android.mk
index d9230e5..168b32a 100644
--- a/Android.mk
+++ b/Android.mk
@@ -393,6 +393,7 @@
         src/armnn/test/optimizations/OptimizeInversePermutesTests.cpp \
         src/armnn/test/optimizations/PermuteAndBatchToSpaceAsDepthToSpaceTests.cpp \
         src/armnn/test/optimizations/PermuteAsReshapeTests.cpp \
+        src/armnn/test/optimizations/ReduceMultipleAxesTests.cpp \
         src/armnn/test/optimizations/SquashEqualSiblingsTests.cpp \
         src/armnn/test/optimizations/TransposeAsReshapeTests.cpp \
         src/armnn/test/OptimizerTests.cpp \