aboutsummaryrefslogtreecommitdiff
path: root/src/armnnTfParser/TfParser.hpp
diff options
context:
space:
mode:
authorFrancisMurtagh <francis.murtagh@arm.com>2019-01-24 10:53:39 +0000
committerFrancis Murtagh <francis.murtagh@arm.com>2019-01-24 12:43:25 +0000
commit94412aff782472be54dce4328e2ecee0225b3e97 (patch)
tree42de97582e1948ecd91bdb24de00a966d180f6e7 /src/armnnTfParser/TfParser.hpp
parent5f4e41ff596170dad9c073b007b3f53783a9e1f3 (diff)
downloadarmnn-94412aff782472be54dce4328e2ecee0225b3e97.tar.gz
IVGCVSW-2512 Add Gather operator parser to TfParser
* Add ParseGather to TFParser * Add Unit tests for Gather Operator !armnn:562 Change-Id: Idff45c2d3d8d683aa9eb2c4a63123c8d6054609e
Diffstat (limited to 'src/armnnTfParser/TfParser.hpp')
-rw-r--r--src/armnnTfParser/TfParser.hpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.hpp b/src/armnnTfParser/TfParser.hpp
index f1b7205ff1..9a7d7827c3 100644
--- a/src/armnnTfParser/TfParser.hpp
+++ b/src/armnnTfParser/TfParser.hpp
@@ -163,6 +163,7 @@ private:
ParsedTfOperationPtr ParseEqual(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef);
ParsedTfOperationPtr ParseMaximum(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef);
ParsedTfOperationPtr ParseMinimum(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef);
+ ParsedTfOperationPtr ParseGather(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef);
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);