diff options
author | Bruno Goncalves <bruno.slackware@gmail.com> | 2018-12-27 14:20:35 -0200 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2019-01-22 17:10:06 +0000 |
commit | 9c761a6335f58901bef01b33f1127f9fae3b2bf3 (patch) | |
tree | 114970cbde18a4facdc655a2bb917fd06390f168 /src/armnnTfLiteParser/TfLiteParser.hpp | |
parent | 33f8e3b6c71070fd867809ca6934069a950081dc (diff) | |
download | armnn-9c761a6335f58901bef01b33f1127f9fae3b2bf3.tar.gz |
Added AddBroadcastReshapeLayer method to TfLiteParser
Change-Id: I6027f6dcdb3ed23505f0a9c780bd3e3d45d3daff
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, |