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

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
+}