From 0fe7376f6982842e32591f3996d5ee650d22156f Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Thu, 20 Aug 2020 15:38:29 +0100 Subject: Bugfix: Allow permutation of QuantizationDim Signed-off-by: Francis Murtagh Change-Id: Ib98ec471e6fdd47600b7c62d0b4d19dd36e20cbd --- include/armnnUtils/Permute.hpp | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'include') diff --git a/include/armnnUtils/Permute.hpp b/include/armnnUtils/Permute.hpp index 4e9bfc0823..d719f4a623 100644 --- a/include/armnnUtils/Permute.hpp +++ b/include/armnnUtils/Permute.hpp @@ -11,9 +11,12 @@ namespace armnnUtils { -armnn::TensorShape Permuted(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings); +armnn::TensorShape Permuted(const armnn::TensorShape& srcShape, + const armnn::PermutationVector& mappings); -armnn::TensorInfo Permuted(const armnn::TensorInfo& info, const armnn::PermutationVector& mappings); +armnn::TensorInfo Permuted(const armnn::TensorInfo& info, + const armnn::PermutationVector& mappings, + bool perChannelPermute = false); void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings, const void* src, void* dst, size_t dataTypeSize); -- cgit v1.2.1