From 8cfcbb22855817e6af7bc2c967d66ec638106d01 Mon Sep 17 00:00:00 2001 From: Ryan OShea Date: Fri, 26 Jun 2020 14:40:29 +0100 Subject: IVGCVSW-4919 Mean VTS/NNT Failures * Allow output to be reduced to scalar Signed-off-by: Ryan OShea Change-Id: I3b5a392fe70a84814c523a22d6fe3cf19d74f83f --- src/armnn/layers/MeanLayer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/armnn/layers') diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp index b39268c7ca..9cefd7d1f8 100644 --- 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"); } -- cgit v1.2.1