diff options
author | Nina Drozd <nina.drozd@arm.com> | 2019-04-09 09:37:38 +0100 |
---|---|---|
committer | Nina Drozd <nina.drozd@arm.com> | 2019-04-09 11:32:39 +0000 |
commit | 998517647d699d602e36f06b40d3f1d1ddaae7be (patch) | |
tree | 4b9505e7476f41ad38960ffc3e1271b32cdea1ad /src/armnnTfLiteParser/TfLiteParser.hpp | |
parent | 2e6dc3a1c5d47825535db7993ba77eb1596ae99b (diff) | |
download | armnn-998517647d699d602e36f06b40d3f1d1ddaae7be.tar.gz |
IVGCVSW-2845: Add TfLite Parser support for TanH activation layer
* Added ParseTanH in TfLiteParser
* Added testcase for parsing TanH activation in Activations.cpp
* Added new supported TanH operator in TensorflowLiteSupport.md
Signed-off-by: Nina Drozd <nina.drozd@arm.com>
Change-Id: Ie60f38994d76b077f86828c21c57381bdeeaeb5a
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 e166dd5d8c..a3ef22fa7e 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -117,6 +117,7 @@ private: void ParsePad(size_t subgraphIndex, size_t operatorIndex); void ParsePool(size_t subgraphIndex, size_t operatorIndex, armnn::PoolingAlgorithm algorithm); void ParseSplit(size_t subgraphIndex, size_t operatorIndex); + void ParseTanH(size_t subgraphIndex, size_t operatorIndex); void RegisterProducerOfTensor(size_t subgraphIndex, size_t tensorIndex, armnn::IOutputSlot* slot); void RegisterConsumerOfTensor(size_t subgraphIndex, size_t tensorIndex, armnn::IInputSlot* slot); |