diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-12-18 22:10:47 +0000 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-12-18 22:10:47 +0000 |
commit | 93cbbcc762769b15dc5474ddda14329bb458fa86 (patch) | |
tree | 778ad1133db0a7dff2a359415d87a74a70018788 /src | |
parent | c374ff06b9e9f45bcf48f4af6bda054f4b243cf6 (diff) | |
download | armnn-93cbbcc762769b15dc5474ddda14329bb458fa86.tar.gz |
IVGCVSW-4302 Depthwise CTS. Fix ReorderWeightChannelsForAcl.
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I8d2050f4478fe9d9cdf9e374b8906827cc769689
Diffstat (limited to 'src')
-rw-r--r-- | src/backends/backendsCommon/WorkloadUtils.cpp | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/src/backends/backendsCommon/WorkloadUtils.cpp b/src/backends/backendsCommon/WorkloadUtils.cpp index 0cf8ad1a94..1a45a9f6e9 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++) { |