diff options
author | Kevin May <kevin.may@arm.com> | 2023-07-04 16:10:55 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2023-07-06 08:57:15 +0000 |
commit | 4a54daadf070c96a481f0da0241a5981b849a0ee (patch) | |
tree | 8aa2945960737d9f59ffd6212c89cdb3278845da /shim | |
parent | f39f3f437cddcedba9a3740fb36d476e652680f5 (diff) | |
download | armnn-4a54daadf070c96a481f0da0241a5981b849a0ee.tar.gz |
IVGCVSW-7846 Fix incorrect Concat permutation params in Support Library
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I17e49ee167a208d9810eedbb0d3a220b66d34aa7
Diffstat (limited to 'shim')
-rw-r--r-- | shim/sl/canonical/ConversionUtils.hpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/shim/sl/canonical/ConversionUtils.hpp b/shim/sl/canonical/ConversionUtils.hpp index 91a8e3080c..f595db2553 100644 --- a/shim/sl/canonical/ConversionUtils.hpp +++ b/shim/sl/canonical/ConversionUtils.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -455,7 +455,7 @@ void SanitizeBiasQuantizationScale(armnn::TensorInfo& biasInfo, // 4D Tensor Permutations const armnn::PermutationVector IdentityPermutation4D({ 0U, 1U, 2U, 3U }); const armnn::PermutationVector IdentityPermutation3D({ 0U, 1U, 2U }); -const armnn::PermutationVector SwapDim1And2({ 0U, 2U, 1U, 3U }); +const armnn::PermutationVector SwapDim2And3({ 0U, 1U, 3U, 2U }); // 3D Permutation Vectors const armnn::PermutationVector RotateTensorLeft({ 1U, 2U, 0U }); @@ -603,8 +603,8 @@ bool CreateConcatPermutationParameters(const unsigned int numberOfDimensions, // or along dimension 0 or 2 for a 3-D tensor. if (numberOfDimensions == 4 && concatDimension == 2) { - concatDimension = 1; - permutationPair = std::make_pair(SwapDim1And2, SwapDim1And2); + concatDimension = 3; + permutationPair = std::make_pair(SwapDim2And3, SwapDim2And3); needPermute = true; } else if (numberOfDimensions == 3 && concatDimension == 1) |