diff options
Diffstat (limited to 'src/armnn/layers/MeanLayer.cpp')
-rw-r--r-- | src/armnn/layers/MeanLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp index 6bbb0943b0..01f1133c5c 100644 --- a/src/armnn/layers/MeanLayer.cpp +++ b/src/armnn/layers/MeanLayer.cpp @@ -60,7 +60,7 @@ void MeanLayer::ValidateTensorShapesFromInputs() { 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"); } |