commit 8cfcbb22855817e6af7bc2c967d66ec638106d01
author Ryan OShea <Ryan.OShea2@arm.com> Fri Jun 26 14:40:29 2020 +0100
committer Ryan O'Shea <ryan.oshea2@arm.com> Fri Jun 26 13:56:59 2020 +0000
tree 3fd5adb0211149b0e9f6530accfb9e48910bcd13
parent cdc0149ffe40f14ff4695149d9bdf551f8e07702

IVGCVSW-4919 Mean VTS/NNT Failures

 * Allow output to be reduced to scalar

Signed-off-by: Ryan OShea <Ryan.OShea2@arm.com>
Change-Id: I3b5a392fe70a84814c523a22d6fe3cf19d74f83f

src/armnn/layers/MeanLayer.cpp

diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp
index b39268c..9cefd7d 100644
--- a/src/armnn/layers/MeanLayer.cpp
+++ b/src/armnn/layers/MeanLayer.cpp
@@ -61,7 +61,7 @@
     {
         outputRank = 1;
     }
-    else if (m_Param.m_Axis.size() >= input.GetNumDimensions())
+    else if (m_Param.m_Axis.size() > input.GetNumDimensions())
     {
         throw LayerValidationException("MeanLayer: Dimensions to reduce can not be bigger than input dimensions");
     }