diff options
author | saoste01 <saoirse.stewart@arm.com> | 2018-08-28 15:41:51 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-17 17:21:25 +0100 |
commit | bbd4061da3094e1eec61fcdb6297f626ff025ba2 (patch) | |
tree | c51c1d2a62e6ccc86c913b5d20b9f3eba314d759 /src/armnnTfParser/TfParser.hpp | |
parent | 8c5e3dc9f04ac15aa926d1a1214610760830dbe2 (diff) | |
download | armnn-bbd4061da3094e1eec61fcdb6297f626ff025ba2.tar.gz |
IVGCVSW-1779: Updating TF Parser to include the RealDiv operator
Change-Id: I4b193d9119f5aaf41081335df043352953364c87
Diffstat (limited to 'src/armnnTfParser/TfParser.hpp')
-rw-r--r-- | src/armnnTfParser/TfParser.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.hpp b/src/armnnTfParser/TfParser.hpp index 75cd3a5bd0..534714cbee 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -136,6 +136,7 @@ private: ParsedTfOperationPtr ParseMatMul(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseMul(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParsePlaceholder(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); + ParsedTfOperationPtr ParseRealDiv(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseRelu(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseRelu6(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseReshape(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); @@ -153,6 +154,7 @@ private: ParsedTfOperationPtr ParseMaximum(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); private: armnn::IConnectableLayer* AddMultiplicationLayer(const tensorflow::NodeDef& nodeDef); |