diff options
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, |