commit e0200da9300ab0586d1d0b8c855ae1679f2084ab
author Pablo Tello <pablo.tello@arm.com> Wed May 29 14:09:19 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Thu May 30 07:52:45 2019 +0000
tree ddc9d413a623ece8fadac8a72fa5350960a9f0a3
parent 248683fef73807b49e5a58fbd0428900b7e9cfb9

MLCE-119: Allow deph multipliers 2 and 3

We had check in the parser to catch calls where deph_mult > 1, this came from
the time when ACL supported only multiplier == 1.

Change-Id: Ic21a2cd6a5bb328d43fc9c667e847429a57760b2
Signed-off-by: Pablo Tello <pablo.tello@arm.com>

src/armnnTfLiteParser/TfLiteParser.cpp

diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index 11d5b5c..d0d130d 100644
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -789,8 +789,8 @@
     desc.m_StrideX = CHECKED_NON_NEGATIVE(options->stride_w);
     desc.m_StrideY = CHECKED_NON_NEGATIVE(options->stride_h);
     desc.m_DataLayout = armnn::DataLayout::NHWC;
-    // ACL only supports a depth (channel) multiplier of 1, it is not currently stored in the descriptor
-    CHECK_VALID_SIZE(CHECKED_NON_NEGATIVE(options->depth_multiplier), 1);
+    // ACL only supports a depth (channel) multiplier of {1,2,3}, it is not currently stored in the descriptor
+    CHECK_VALID_SIZE(CHECKED_NON_NEGATIVE(options->depth_multiplier), 1,2,3 );
 
     auto inputs = GetInputs(m_Model, subgraphIndex, operatorIndex);
     CHECK_VALID_SIZE(inputs.size(), 2, 3);