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/CaffeSupport.md | |
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/CaffeSupport.md')
-rw-r--r-- | src/armnnCaffeParser/CaffeSupport.md | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnnCaffeParser/CaffeSupport.md b/src/armnnCaffeParser/CaffeSupport.md index ab0d3a4ab8..3501a78ae8 100644 --- a/src/armnnCaffeParser/CaffeSupport.md +++ b/src/armnnCaffeParser/CaffeSupport.md @@ -16,8 +16,10 @@ Although some other neural networks might work, Arm tests the Arm NN SDK with Ca The Arm NN SDK supports the following machine learning layers for Caffe networks: +- Argmax, excluding the top_k and out_max_val parameters. - BatchNorm, in inference mode. -- Convolution, excluding the Dilation Size, Weight Filler, Bias Filler, Engine, Force nd_im2col, and Axis parameters. +- Convolution, excluding Weight Filler, Bias Filler, Engine, Force nd_im2col, and Axis parameters. +- Deconvolution, excluding the Dilation Size, Weight Filler, Bias Filler, Engine, Force nd_im2col, and Axis parameters. Caffe doesn't support depthwise convolution, the equivalent layer is implemented through the notion of groups. ArmNN supports groups this way: - when group=1, it is a normal conv2d @@ -35,4 +37,4 @@ The Arm NN SDK supports the following machine learning layers for Caffe networks - Softmax, excluding the Axis and Engine parameters. - Split. -More machine learning layers will be supported in future releases.
\ No newline at end of file +More machine learning layers will be supported in future releases. |