From 77707452ce461a67c27f54ecf4e340759acf834f Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Wed, 18 Dec 2019 22:10:47 +0000 Subject: IVGCVSW-4302 Depthwise CTS. Fix ReorderWeightChannelsForAcl. Signed-off-by: Teresa Charlin Change-Id: I8d2050f4478fe9d9cdf9e374b8906827cc769689 --- src/backends/backendsCommon/WorkloadUtils.cpp | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/src/backends/backendsCommon/WorkloadUtils.cpp b/src/backends/backendsCommon/WorkloadUtils.cpp index 3abadd7563..abb704e80a 100644 --- a/src/backends/backendsCommon/WorkloadUtils.cpp +++ b/src/backends/backendsCommon/WorkloadUtils.cpp @@ -86,17 +86,12 @@ ConstTensor ReorderWeightChannelsForAcl(const ConstTensor& weightHandle, DataLay 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++) { -- cgit v1.2.1