commit 5786eb7b4021ebed55e3fca31cb2594b81472cb5
author Teresa Charlin <teresa.charlinreyes@arm.com> Fri May 21 16:29:45 2021 +0100
committer Jim Flynn <jim.flynn@arm.com> Mon May 24 14:39:11 2021 +0000
tree 2fefea3e753b1c0f81742a854429b91ea9f76007
parent cb0f6107e7de2bbb5ab8f06e80261fa091e0bfa6

IVGCVSW-6069 Fold PAD into Depthwise Convolution

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ib01629256309cfe17f341909d5b9bbbb09361422

src/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index f097e67..2c14136 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1610,6 +1610,7 @@
                                                 TransposeAsReshape(),
                                                 OptimizeConsecutiveReshapes(),
                                                 FoldPadIntoConvolution2d(),
+                                                FoldPadIntoDepthwiseConvolution2d(),
                                                 FoldPadIntoPooling2d(),
                                                 PermuteAndBatchToSpaceAsDepthToSpace(),
                                                 TransposeAndBatchToSpaceAsDepthToSpace(),