From e4112637b5057f638c3d147a1751ed01772be48a Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Wed, 8 Jul 2020 11:12:47 +0100 Subject: IVGCVSW-5087 NNT zero sized failing after TensorShape refactor Signed-off-by: Teresa Charlin Change-Id: I71bf28aae86766db8777b011c769b547436e7d39 --- src/armnn/Tensor.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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]; -- cgit v1.2.1