projects
/
platform
/
upstream
/
armnn.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
cdc0149
)
IVGCVSW-4919 Mean VTS/NNT Failures
author
Ryan OShea
<Ryan.OShea2@arm.com>
Fri, 26 Jun 2020 13:40:29 +0000
(14:40 +0100)
committer
Ryan O'Shea
<ryan.oshea2@arm.com>
Fri, 26 Jun 2020 13:56:59 +0000
(13:56 +0000)
* Allow output to be reduced to scalar
Signed-off-by: Ryan OShea <Ryan.OShea2@arm.com>
Change-Id: I3b5a392fe70a84814c523a22d6fe3cf19d74f83f
src/armnn/layers/MeanLayer.cpp
patch
|
blob
|
history
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");
}