aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Network.cpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2021-05-21 16:29:45 +0100
committerJim Flynn <jim.flynn@arm.com>2021-05-24 14:39:11 +0000
commit5786eb7b4021ebed55e3fca31cb2594b81472cb5 (patch)
tree2fefea3e753b1c0f81742a854429b91ea9f76007 /src/armnn/Network.cpp
parentcb0f6107e7de2bbb5ab8f06e80261fa091e0bfa6 (diff)
downloadarmnn-5786eb7b4021ebed55e3fca31cb2594b81472cb5.tar.gz
IVGCVSW-6069 Fold PAD into Depthwise Convolution
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: Ib01629256309cfe17f341909d5b9bbbb09361422
Diffstat (limited to 'src/armnn/Network.cpp')
-rw-r--r--src/armnn/Network.cpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index f097e677d7..2c14136080 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1610,6 +1610,7 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork,
TransposeAsReshape(),
OptimizeConsecutiveReshapes(),
FoldPadIntoConvolution2d(),
+ FoldPadIntoDepthwiseConvolution2d(),
FoldPadIntoPooling2d(),
PermuteAndBatchToSpaceAsDepthToSpace(),
TransposeAndBatchToSpaceAsDepthToSpace(),