commit 4b19d2249e3b8f9216ec5b410fad20c41b4c6053
author Cathal Corbett <cathal.corbett@arm.com> Wed May 11 20:12:17 2022 +0100
committer Cathal Corbett <cathal.corbett@arm.com> Thu May 12 13:41:56 2022 +0100
tree b7511932f11b2c27f39e1e76ac290ba3baca0fda
parent 50bc39e401532f9aa838e3d9b1f3e3ae5845e37f

IVGCVSW-6940 ConstTensorsAsInput: DepthwiseConvolution2d - Complete ACL

  * Added backend specific optimization & test for CpuAcc and GpuAcc: PermuteDepthwiseConv2dWeights

Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I600476b2e9c557a39818a574c1091c9d650b21b1

src/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 549222b..d2ebd4c 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1167,6 +1167,11 @@
         auto backendObjPtr = backends.find(selectedBackend)->second.get();
         ARMNN_ASSERT(backendObjPtr);
 
+        if(selectedBackend == armnn::Compute::GpuAcc || selectedBackend == armnn::Compute::CpuAcc)
+        {
+            Optimizer::Pass(optGraph, MakeOptimizations(optimizations::PermuteDepthwiseConv2dWeights()));
+        }
+
         // Select sub-graphs based on backend
         SubgraphViewSelector::Subgraphs subgraphs =
                 SubgraphViewSelector::SelectSubgraphs(optGraph,