IVGCVSW-4302 Depthwise CTS. Fix ReorderWeightChannelsForAcl.
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I8d2050f4478fe9d9cdf9e374b8906827cc769689
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++)
{