diff options
author | Pablo Tello <pablo.tello@arm.com> | 2019-05-29 14:09:19 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-30 07:52:45 +0000 |
commit | e0200da9300ab0586d1d0b8c855ae1679f2084ab (patch) | |
tree | ddc9d413a623ece8fadac8a72fa5350960a9f0a3 /src | |
parent | 248683fef73807b49e5a58fbd0428900b7e9cfb9 (diff) | |
download | armnn-e0200da9300ab0586d1d0b8c855ae1679f2084ab.tar.gz |
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>
Diffstat (limited to 'src')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 4 |
1 files 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); |