diff options
| -rw-r--r-- | 1.1/HalPolicy.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/1.1/HalPolicy.cpp b/1.1/HalPolicy.cpp index 64b7d2ae..5530d310 100644 --- a/1.1/HalPolicy.cpp +++ b/1.1/HalPolicy.cpp @@ -237,7 +237,7 @@ bool HalPolicy::ConvertPad(const Operation& operation, const Model& model, Conve 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); } |