diff options
| author | Bruno Goncalves <bruno.slackware@gmail.com> | 2019-02-08 18:52:21 -0200 |
|---|---|---|
| committer | derek.lamberti <derek.lamberti@arm.com> | 2019-02-22 15:29:25 +0000 |
| commit | db947e26cf40b7dd274be3741f4f5f9231c10cef (patch) | |
| tree | 2f2612c87d0b53a6168d2a521dedd51e42ca588b /src/armnnTfLiteParser/TfLiteParser.hpp | |
| parent | 3f58ddb30362a927f3ff63c59545382bef77c25f (diff) | |
| download | armnn-db947e26cf40b7dd274be3741f4f5f9231c10cef.tar.gz | |
Add batch-to-space-nd parser to tf-lite
Change-Id: I1e210f88f19f38ab719008e119c8fed153a3dd95
Signed-off-by: Bruno Goncalves <bruno.slackware@gmail.com>
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.hpp')
| -rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp index 599923244e..ceca9e2c44 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -92,6 +92,7 @@ private: void ParseUnsupportedOperator(size_t subgraphIndex, size_t operatorIndex); void ParseActivation(size_t subgraphIndex, size_t operatorIndex, armnn::ActivationFunction activationType); void ParseAveragePool2D(size_t subgraphIndex, size_t operatorIndex); + void ParseBatchToSpaceND(size_t subgraphIndex, size_t operatorIndex); void ParseConcatenation(size_t subgraphIndex, size_t operatorIndex); void ParseConv2D(size_t subgraphIndex, size_t operatorIndex); void ParseDepthwiseConv2D(size_t subgraphIndex, size_t operatorIndex); |