diff options
author | jimfly01 <jim.flynn@arm.com> | 2018-12-04 10:09:52 +0000 |
---|---|---|
committer | jimfly01 <jim.flynn@arm.com> | 2018-12-05 12:15:48 +0000 |
commit | f6ba747c0802d87ba30aecd598f0603f9bd18576 (patch) | |
tree | df892f37ed1edc13e9ed05d9ac72f67a1fd622a3 /src/armnnTfParser/TfParser.hpp | |
parent | 2135015779092e259ad5d5df185eda0c34b56359 (diff) | |
download | armnn-f6ba747c0802d87ba30aecd598f0603f9bd18576.tar.gz |
IVGCVSW-2296 Add ParsePad method to TfParser
* Also added unit test armnnTfParser/test/Pad.cpp
* Added missing 'Pad' entry to GetLayerTypeAsCString(LayerType)
* Fixed the RefLayerSupport.IsPadSupported, now it returns true
* Small fix in Optimize. Only resolve stringstream to string once
Change-Id: Ieaa1886858a48cd761ac5f30454f73e44bdd4b8f
Diffstat (limited to 'src/armnnTfParser/TfParser.hpp')
-rw-r--r-- | src/armnnTfParser/TfParser.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.hpp b/src/armnnTfParser/TfParser.hpp index 7abf783c38..1c29ce2717 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -124,6 +124,8 @@ private: /// Checks if there is a pre-parsed const tensor available with the given name and Type. template<typename Type> bool HasParsedConstTensor(const std::string & nodeName) const; + template<typename Type> + bool HasParsedConstTensor(ParsedTfOperation* parsedTfOpPtr) const; ParsedTfOperationPtr ParseAdd(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseBiasAdd(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); @@ -153,6 +155,7 @@ private: armnn::PoolingAlgorithm pooltype); ParsedTfOperationPtr ParseMaximum(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseMinimum(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); + ParsedTfOperationPtr ParsePad(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); |