From c02d4411b6bf9d755e25a1593c9fe16e7d93901e Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Tue, 18 Feb 2020 15:18:27 +0000 Subject: 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 Change-Id: I243b33fda2762180ef880458e760a54fde64f3d8 --- src/armnnTfParser/TfParser.hpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/armnnTfParser/TfParser.hpp') 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); -- cgit v1.2.1