diff options
author | Keith Mok <ek9852@gmail.com> | 2020-12-20 19:47:25 -0800 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2021-01-07 14:01:15 +0000 |
commit | 9801b17f9d0786ce107119af88b92a654e04396f (patch) | |
tree | 6abc04602bdb309561c6b6fa6242f535e9add3f9 /src/armnnCaffeParser/CaffeParser.hpp | |
parent | 7dc18207d86299cf605a2316e86dc9c098708729 (diff) | |
download | armnn-9801b17f9d0786ce107119af88b92a654e04396f.tar.gz |
Add argmax deconv support for caffe parser
armnn support argmax and deconv , but caffe parser does not.
Add back this feature.
Signed-off-by: Keith Mok <ek9852@gmail.com>
Change-Id: I6b99cc4b58491204c41c6e1d11f583c65c628ee4
Diffstat (limited to 'src/armnnCaffeParser/CaffeParser.hpp')
-rw-r--r-- | src/armnnCaffeParser/CaffeParser.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/armnnCaffeParser/CaffeParser.hpp b/src/armnnCaffeParser/CaffeParser.hpp index 105204e923..98eeffc6a1 100644 --- a/src/armnnCaffeParser/CaffeParser.hpp +++ b/src/armnnCaffeParser/CaffeParser.hpp @@ -56,6 +56,7 @@ protected: /// @{ void ParseInputLayer(const caffe::LayerParameter& layerParam); void ParseConvLayer(const caffe::LayerParameter& layerParam); + void ParseDeconvLayer(const caffe::LayerParameter& layerParam); void ParsePoolingLayer(const caffe::LayerParameter& layerParam); void ParseReluLayer(const caffe::LayerParameter& layerParam); void ParseLRNLayer(const caffe::LayerParameter& layerParam); @@ -67,6 +68,7 @@ protected: void ParseScaleLayer(const caffe::LayerParameter& layerParam); void ParseSplitLayer(const caffe::LayerParameter& layerParam); void ParseDropoutLayer(const caffe::LayerParameter& layerParam); + void ParseArgmaxLayer(const caffe::LayerParameter& layerParam); /// @} /// ParseConv may use these helpers depending on the group parameter @@ -79,6 +81,10 @@ protected: const armnn::Convolution2dDescriptor & desc, unsigned int kernelW, unsigned int kernelH); + void AddDeconvLayerWithSplits(const caffe::LayerParameter& layerParam, + const armnn::TransposeConvolution2dDescriptor& desc, + unsigned int kernelW, + unsigned int kernelH); /// @} /// Converts Caffe's protobuf tensor shape format to ArmNN's @@ -170,4 +176,4 @@ public: CaffeParser(); }; -}
\ No newline at end of file +} |