diff options
author | Mike Kelly <mike.kelly@arm.com> | 2023-01-19 18:29:40 +0000 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2023-01-24 22:57:16 +0000 |
commit | 04d8229bb3e78d1b1dd21eed41e47aabc25d8e2f (patch) | |
tree | 478dbaf477eaa59fac838e6e73b56843f80b52d0 /src/armnnTfLiteParser/TfLiteParser.hpp | |
parent | 0e3fe10bfe1b4f006f6e0c5c2fae8fb5515c7544 (diff) | |
download | armnn-04d8229bb3e78d1b1dd21eed41e47aabc25d8e2f.tar.gz |
IVGCVSW-7277 Fixed issues with FullyConnected on certain TFLite models
* TFLite Parser:
* Fixed issue in ParseReshape where the targetShape wasn't always calculated correctly
* Fixed issue in ParseFullyConnected where the wrong name was used for the ReshapeLayer
* Added an ExpandDims to the FullyConnected to ensure that we reshape the output correctly
* TFLite Delegate:
* Added an ExpandDims to the FullyConnected to ensure that we reshape the output correctly
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I129dfcb8543f8a3a297c0589c841be20ef3b6407
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.hpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp index cf334cc70c..327b8a8651 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2018-2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -218,6 +218,12 @@ private: size_t operatorIndex, armnn::IConnectableLayer* layer); + /// Attach an reshape layer to the one passed as a parameter + armnn::IConnectableLayer* AddReshapeLayer(armnn::IConnectableLayer* layer, + unsigned int outputSlot, + std::string reshapeLayerName, + armnn::TensorInfo outputShape); + /// Attach an activation layer to the one passed as a parameter armnn::IConnectableLayer* AddFusedActivationLayer(armnn::IConnectableLayer* layer, unsigned int outputSlot, |