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

diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp
index b39268c..9cefd7d 100644 (file)
--- a/src/armnn/layers/MeanLayer.cpp
+++ b/src/armnn/layers/MeanLayer.cpp
@@ -61,7 +61,7 @@ void MeanLayer::ValidateTensorShapesFromInputs(ShapeInferenceMethod shapeInferen
     {
         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");
     }