diff options
| author | Conor Kennedy <conor.kennedy@arm.com> | 2018-12-05 11:05:54 +0000 |
|---|---|---|
| committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2018-12-05 13:36:01 +0000 |
| commit | c2130a070e6a9196d193c93a02b5f118810dd59a (patch) | |
| tree | 45994da8a86619e4ed942a6619df284954329a1f /src/armnnTfParser/TfParser.hpp | |
| parent | f6ba747c0802d87ba30aecd598f0603f9bd18576 (diff) | |
| download | armnn-c2130a070e6a9196d193c93a02b5f118810dd59a.tar.gz | |
IVGCVSW-2193 ExpandDims operation implementation
* Add ExpandDims operation to TfParser.cpp
Change-Id: Ifa756ae0667c11e3b6daec8f6dd4e54cac88d16a
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 1c29ce2717..da78f48f54 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -131,6 +131,7 @@ private: ParsedTfOperationPtr ParseBiasAdd(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseConv2D(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseDepthwiseConv2D(const tensorflow::NodeDef& nodeDef,const tensorflow::GraphDef& graphDef); + ParsedTfOperationPtr ParseExpandDims(const tensorflow::NodeDef& nodeDef,const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseFusedBatchNorm(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseConcat(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseIdentity(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); |