diff options
author | Éanna Ó Catháin <eanna.ocathain@arm.com> | 2019-01-14 16:18:49 +0000 |
---|---|---|
committer | Éanna Ó Catháin <eanna.ocathain@arm.com> | 2019-01-14 16:18:49 +0000 |
commit | 074f1ec6063a692ffacd8f0c0d132216a98ae509 (patch) | |
tree | 87c7f8cba95100f30348faea6d0d1cb7f7d609fc /1.1/HalPolicy.cpp | |
parent | 265d1ad1c66eefc5c45025c314e2e2ac49a15949 (diff) | |
download | android-nn-driver-074f1ec6063a692ffacd8f0c0d132216a98ae509.tar.gz |
MLCE-86 Fixing up the problem with the invalid padding clause
Change-Id: I34d8a7f359acc19258a9d6e7f9256f8b33777e20
Diffstat (limited to '1.1/HalPolicy.cpp')
-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); } |