diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-02-18 15:18:27 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-02-18 17:38:34 +0000 |
commit | 48d709356269f4f4131d319f6638ab4f7b083931 (patch) | |
tree | cd354f9b1294087777cb43a891f0d099c179e39a /src/armnnTfParser/TfParser.hpp | |
parent | 98d39da80a3376df52e2bd3d77a2691f0e0ec209 (diff) | |
download | armnn-48d709356269f4f4131d319f6638ab4f7b083931.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); |