aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2020-07-08 11:12:47 +0100
committerTeresa Charlin <teresa.charlinreyes@arm.com>2020-07-08 11:12:47 +0100
commite4112637b5057f638c3d147a1751ed01772be48a (patch)
treecd41b355dc9051dec5fe06667c5bf3ad83c9103e
parentaaf9a8fadce26375042d81bb241e36ccdcda0874 (diff)
downloadarmnn-e4112637b5057f638c3d147a1751ed01772be48a.tar.gz
IVGCVSW-5087 NNT zero sized failing after TensorShape refactor
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: I71bf28aae86766db8777b011c769b547436e7d39
-rw-r--r--src/armnn/Tensor.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/Tensor.cpp b/src/armnn/Tensor.cpp
index 6ad46c59c9..80aa0a920f 100644
--- a/src/armnn/Tensor.cpp
+++ b/src/armnn/Tensor.cpp
@@ -192,7 +192,7 @@ unsigned int TensorShape::GetNumElements() const
bool atLeastOneDimensionSpecified = false;
for (unsigned int i = 0; i < m_NumDimensions; ++i)
{
- if (m_DimensionsSpecificity[i] && m_Dimensions[i] != 0)
+ if (m_DimensionsSpecificity[i])
{
atLeastOneDimensionSpecified = true;
count *= m_Dimensions[i];