commit 074f1ec6063a692ffacd8f0c0d132216a98ae509
author Éanna Ó Catháin <eanna.ocathain@arm.com> Mon Jan 14 16:18:49 2019 +0000
committer Éanna Ó Catháin <eanna.ocathain@arm.com> Mon Jan 14 16:18:49 2019 +0000
tree 87c7f8cba95100f30348faea6d0d1cb7f7d609fc
parent 265d1ad1c66eefc5c45025c314e2e2ac49a15949

MLCE-86 Fixing up the problem with the invalid padding clause

Change-Id: I34d8a7f359acc19258a9d6e7f9256f8b33777e20

1.1/HalPolicy.cpp

diff --git a/1.1/HalPolicy.cpp b/1.1/HalPolicy.cpp
index 64b7d2a..5530d31 100644
--- a/1.1/HalPolicy.cpp
+++ b/1.1/HalPolicy.cpp
@@ -237,7 +237,7 @@
 
     unsigned int rank = inputInfo.GetNumDimensions();
     armnn::TensorShape paddingsOperandShape = GetTensorShapeForOperand(*paddingsOperand);
-    if (paddingsOperandShape.GetNumDimensions() != rank || paddingsOperandShape.GetNumElements() != 2)
+    if (paddingsOperandShape.GetNumDimensions() != 2 || paddingsOperandShape.GetNumElements() != rank * 2)
     {
         return Fail("%s: Operation has invalid paddings operand: expected shape [%d, 2]",  __func__, rank);
     }