diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2020-08-20 15:38:29 +0100 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-08-21 15:56:13 +0100 |
commit | 186c21c9a598cbe2e81ad53e5b5fd96d75f981f5 (patch) | |
tree | 45cb5522dcf36256f66bbe25eb8b24460cbdb37a /include/armnnUtils/Permute.hpp | |
parent | c84e45d933a9b45810a3bb88f6873f4eddca0975 (diff) | |
download | armnn-186c21c9a598cbe2e81ad53e5b5fd96d75f981f5.tar.gz |
Bugfix: Allow permutation of QuantizationDim
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: Ib98ec471e6fdd47600b7c62d0b4d19dd36e20cbd
Diffstat (limited to 'include/armnnUtils/Permute.hpp')
-rw-r--r-- | include/armnnUtils/Permute.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
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); |