diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-28 11:35:47 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-29 10:30:58 +0000 |
commit | e011d20f279e6e67e899bc6930b0266fc357bc1c (patch) | |
tree | 4a190b08df3c63c4d540506491d31bee7c8dcd46 /include/armnnUtils/Permute.hpp | |
parent | 2e259276fba9fa5c6c2e146de3b26e3d6c6cccc6 (diff) | |
download | armnn-e011d20f279e6e67e899bc6930b0266fc357bc1c.tar.gz |
IVGCVSW-4209 Create a public API for the ArmNN Utils
* Moved the relevant armnnUtils headers to the new location:
include/armnnUtils
* Update the header usage throughout the source code
!android-nn-driver:2387
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Change-Id: I2ba15cebcacafad2b5a1a7b9c3312ffc585e09d6
Diffstat (limited to 'include/armnnUtils/Permute.hpp')
-rw-r--r-- | include/armnnUtils/Permute.hpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/include/armnnUtils/Permute.hpp b/include/armnnUtils/Permute.hpp new file mode 100644 index 0000000000..1ae00a136b --- /dev/null +++ b/include/armnnUtils/Permute.hpp @@ -0,0 +1,21 @@ +// +// Copyright © 2019 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include <armnn/TensorFwd.hpp> +#include <armnn/Types.hpp> + +namespace armnnUtils +{ + +armnn::TensorShape Permuted(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings); + +armnn::TensorInfo Permuted(const armnn::TensorInfo& info, const armnn::PermutationVector& mappings); + +void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings, + const void* src, void* dst, size_t dataTypeSize); + +} // namespace armnnUtils |