From 6bc85258b0a87cde2e1e644d930ea546b6c3a1cc Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Tue, 6 Dec 2022 20:43:06 +0000 Subject: IVGCVSW-6853 Rewrite BuildArmComputePermutationVector() * Some pemutation vectors were not converted correctly. * Add Transpose end to end test. * Comments added with an example to clarify the differences betweeen Transpose and Permute Signed-off-by: Teresa Charlin Change-Id: I6c0954ca6ce00ef5f2a6f3625abe6f4fd27b5cdf --- src/backends/aclCommon/ArmComputeTensorUtils.hpp | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) (limited to 'src/backends/aclCommon/ArmComputeTensorUtils.hpp') diff --git a/src/backends/aclCommon/ArmComputeTensorUtils.hpp b/src/backends/aclCommon/ArmComputeTensorUtils.hpp index fdcd867bba..6ddecf2aaa 100644 --- a/src/backends/aclCommon/ArmComputeTensorUtils.hpp +++ b/src/backends/aclCommon/ArmComputeTensorUtils.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -66,10 +66,14 @@ arm_compute::Pooling3dLayerInfo BuildArmComputePooling3dLayerInfo(const Pooling3 arm_compute::NormalizationLayerInfo BuildArmComputeNormalizationLayerInfo(const NormalizationDescriptor& desc); /// Utility function used to setup an arm_compute::PermutationVector object from an armnn::PermutationVector. -arm_compute::PermutationVector BuildArmComputePermutationVector(const armnn::PermutationVector& vector); +/// \param perm PermutationVector used in Arm NN Permute layer +/// \return PermutationVector used in ACL Transpose layer +arm_compute::PermutationVector BuildArmComputePermutationVector(const armnn::PermutationVector& perm); /// Utility function used to setup an arm_compute::PermutationVector object from an armnn::PermutationVector. -arm_compute::PermutationVector BuildArmComputeTransposeVector(const armnn::PermutationVector& vector); +/// \param perm PermutationVector used in Arm NN Transpose layer +/// \return PermutationVector used in ACL Transpose layer +arm_compute::PermutationVector BuildArmComputeTransposeVector(const armnn::PermutationVector& perm); /// Utility function used to setup an arm_compute::Size2D object from width and height values. arm_compute::Size2D BuildArmComputeSize2D(const unsigned int width, const unsigned int height); -- cgit v1.2.1