From e0200da9300ab0586d1d0b8c855ae1679f2084ab Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Wed, 29 May 2019 14:09:19 +0100 Subject: 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 --- src/armnnTfLiteParser/TfLiteParser.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 11d5b5c93e..d0d130db82 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -789,8 +789,8 @@ void TfLiteParser::ParseDepthwiseConv2D(size_t subgraphIndex, size_t operatorInd 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); -- cgit v1.2.1