diff options
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.hpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp index 6c264372ba..34ae07f392 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -124,6 +124,10 @@ private: void ResetParser(); + void AddBroadcastReshapeLayer(size_t subgraphIndex, + size_t operatorIndex, + armnn::IConnectableLayer* layer); + /// Attach an activation layer to the one passed as a parameter armnn::IConnectableLayer* AddFusedActivationLayer(armnn::IConnectableLayer* layer, unsigned int outputSlot, |