From 074f1ec6063a692ffacd8f0c0d132216a98ae509 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89anna=20=C3=93=20Cath=C3=A1in?= Date: Mon, 14 Jan 2019 16:18:49 +0000 Subject: MLCE-86 Fixing up the problem with the invalid padding clause Change-Id: I34d8a7f359acc19258a9d6e7f9256f8b33777e20 --- 1.1/HalPolicy.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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); } -- cgit v1.2.1