diff options
author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-02-18 11:27:35 +0000 |
---|---|---|
committer | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-02-24 13:29:57 +0000 |
commit | dd3f71b64072c44cec65a7a883d0c3a29659645c (patch) | |
tree | d48b0508811e88d47b4b95a1777742312b2ba4db /src/armnnTfParser/TfParser.hpp | |
parent | cf34f510b891b87b4c7be5edb30272b36cab7b51 (diff) | |
download | armnn-dd3f71b64072c44cec65a7a883d0c3a29659645c.tar.gz |
COMPMID-3060: Add TF Parser support for Transpose
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Change-Id: I9661787071554b38c5b0ab3c98431f3863b98520
Diffstat (limited to 'src/armnnTfParser/TfParser.hpp')
-rw-r--r-- | src/armnnTfParser/TfParser.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.hpp b/src/armnnTfParser/TfParser.hpp index 9277d44cb1..94499ea52d 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -171,6 +171,7 @@ private: ParsedTfOperationPtr ParsePad(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseSub(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseStack(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); + ParsedTfOperationPtr ParseTranspose(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr AddActivationLayer(const tensorflow::NodeDef& nodeDef, armnn::ActivationDescriptor& desc); ParsedTfOperationPtr AddAdditionLayer(const tensorflow::NodeDef& nodeDef, bool isBiasAdd = false); ParsedTfOperationPtr AddRealDivLayer(const tensorflow::NodeDef& nodeDef); |