diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-02-18 15:18:27 +0000 |
---|---|---|
committer | James Conroy <james.conroy@arm.com> | 2020-02-20 14:14:01 +0000 |
commit | c02d4411b6bf9d755e25a1593c9fe16e7d93901e (patch) | |
tree | cd354f9b1294087777cb43a891f0d099c179e39a /src/armnnTfParser/TfParser.hpp | |
parent | 5a0b111c5ba010d87dc535d15b454b062cb50ced (diff) | |
download | armnn-c02d4411b6bf9d755e25a1593c9fe16e7d93901e.tar.gz |
IVGCVSW-2232 Pack/Stack operator support in Tensorflow Parser
* Parser support and unit tests added for Stack opertor in TfParser
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I243b33fda2762180ef880458e760a54fde64f3d8
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 a7d02be33d..9277d44cb1 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -170,6 +170,7 @@ private: ParsedTfOperationPtr ParseGreater(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); 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 AddActivationLayer(const tensorflow::NodeDef& nodeDef, armnn::ActivationDescriptor& desc); ParsedTfOperationPtr AddAdditionLayer(const tensorflow::NodeDef& nodeDef, bool isBiasAdd = false); ParsedTfOperationPtr AddRealDivLayer(const tensorflow::NodeDef& nodeDef); |