diff options
author | Matthew Jackson <matthew.jackson@arm.com> | 2019-07-16 11:39:21 +0100 |
---|---|---|
committer | Matthew Jackson <matthew.jackson@arm.com> | 2019-07-17 15:00:07 +0000 |
commit | bcca1f4a7ba8364f7b5e58e8e8866ccd7d530f92 (patch) | |
tree | 18211014a9d196e35435c44cd8a4e25810971fae /src/armnnTfLiteParser/TfLiteParser.hpp | |
parent | 81e601c5a5ebf3de3dd6418942708158de50252a (diff) | |
download | armnn-bcca1f4a7ba8364f7b5e58e8e8866ccd7d530f92.tar.gz |
IVGCVSW-3423 Add TfLite parser support for Stack (Pack) layer
* Added ParsePack method
* New unit test Pack.cpp
* Updated TensorFlowLiteSupport.md with new supported operator
Signed-off-by: Matthew Jackson <matthew.jackson@arm.com>
Change-Id: I2310b33ee26959b036bb4452a25c90cc1d4cbf20
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 07a0f26590..437e459732 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -105,6 +105,7 @@ private: void ParseMean(size_t subgraphIndex, size_t operatorIndex); void ParseMinimum(size_t subgraphIndex, size_t operatorIndex); void ParseMul(size_t subgraphIndex, size_t operatorIndex); + void ParsePack(size_t subgraphIndex, size_t operatorIndex); void ParsePad(size_t subgraphIndex, size_t operatorIndex); void ParsePool(size_t subgraphIndex, size_t operatorIndex, armnn::PoolingAlgorithm algorithm); void ParseRelu(size_t subgraphIndex, size_t operatorIndex); |