diff options
| author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-01-11 19:29:18 +0000 |
|---|---|---|
| committer | Ferran Balaguer Arm <ferran.balaguer@arm.com> | 2019-01-14 09:45:15 +0000 |
| commit | 51dd62f5725e8a97f3f6957fbc2b899493eb7bb3 (patch) | |
| tree | f8cce612850d49d798686cce5ad2ab7545b6e0b7 /src/armnnTfParser/TfParser.hpp | |
| parent | 992d6dc57d8463729910b688f0fb5825d0d3ccf2 (diff) | |
| download | armnn-51dd62f5725e8a97f3f6957fbc2b899493eb7bb3.tar.gz | |
IVGCVSW-1656 Add Mean support to Tf Parser
Change-Id: I3d31d6b72be1984acdb51fd9e7b5488a7aa5d832
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 4421768fc5..f1b7205ff1 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -140,6 +140,7 @@ private: ParsedTfOperationPtr ParseIdentity(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseLrn(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseMatMul(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); + ParsedTfOperationPtr ParseMean(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); @@ -260,4 +261,5 @@ private: /// Maps output layer names to their corresponding ids and tensor info. std::unordered_map<std::string, BindingPointInfo> m_NetworkOutputsBindingInfo; }; + } |