commit 93cbbcc762769b15dc5474ddda14329bb458fa86
author Teresa Charlin <teresa.charlinreyes@arm.com> Wed Dec 18 22:10:47 2019 +0000
committer Teresa Charlin <teresa.charlinreyes@arm.com> Wed Dec 18 22:10:47 2019 +0000
tree 778ad1133db0a7dff2a359415d87a74a70018788
parent c374ff06b9e9f45bcf48f4af6bda054f4b243cf6

IVGCVSW-4302  Depthwise CTS. Fix ReorderWeightChannelsForAcl.

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

src/backends/backendsCommon/WorkloadUtils.cpp

diff --git a/src/backends/backendsCommon/WorkloadUtils.cpp b/src/backends/backendsCommon/WorkloadUtils.cpp
index 0cf8ad1..1a45a9f 100644
--- a/src/backends/backendsCommon/WorkloadUtils.cpp
+++ b/src/backends/backendsCommon/WorkloadUtils.cpp
@@ -86,17 +86,12 @@
     unsigned int destinationWeightsChannel;
     unsigned int totalChannels = inputChannels * multiplier;
     unsigned int channelSize   = height * width;
+    unsigned int inputChannel  = 0;
 
     for (unsigned int originWeightsChannel = 0; originWeightsChannel < totalChannels; originWeightsChannel++)
     {
-        if (originWeightsChannel % inputChannels == 0)
-        {
-            destinationWeightsChannel = originWeightsChannel / inputChannels;
-        }
-        else
-        {
-            destinationWeightsChannel = (originWeightsChannel - 1) / inputChannels + multiplier;
-        }
+        inputChannel = originWeightsChannel % inputChannels;
+        destinationWeightsChannel = (originWeightsChannel - inputChannel) / inputChannels + multiplier * inputChannel;
 
         for (unsigned int i = 0; i < channelSize; i++)
         {