diff options
author | Bruno Goncalves <bruno.slackware@gmail.com> | 2019-02-08 19:02:48 -0200 |
---|---|---|
committer | derek.lamberti <derek.lamberti@arm.com> | 2019-02-22 15:29:36 +0000 |
commit | baded14fc01be6c529c4755cda0cb39a278f0b39 (patch) | |
tree | 19765671c7da05bf976404b6c9869a17906ff8ad /src/armnnTfLiteParser/TfLiteParser.hpp | |
parent | db947e26cf40b7dd274be3741f4f5f9231c10cef (diff) | |
download | armnn-baded14fc01be6c529c4755cda0cb39a278f0b39.tar.gz |
Add space-to-batch-nd parser to tf-lite
Change-Id: I3bf86d44f811380559ec35eed0bc43b3bd97da80
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 ceca9e2c44..e074b765f1 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -105,6 +105,7 @@ private: void ParseReshape(size_t subgraphIndex, size_t operatorIndex); void ParseResizeBilinear(size_t subgraphIndex, size_t operatorIndex); void ParseSoftmax(size_t subgraphIndex, size_t operatorIndex); + void ParseSpaceToBatchND(size_t subgraphIndex, size_t operatorIndex); void ParseSqueeze(size_t subgraphIndex, size_t operatorIndex); void ParseSub(size_t subgraphIndex, size_t operatorIndex); void ParseAdd(size_t subgraphIndex, size_t operatorIndex); |