diff options
Diffstat (limited to 'include/armnnUtils/Transpose.hpp')
-rw-r--r-- | include/armnnUtils/Transpose.hpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/include/armnnUtils/Transpose.hpp b/include/armnnUtils/Transpose.hpp new file mode 100644 index 0000000000..0a1ba7dfc1 --- /dev/null +++ b/include/armnnUtils/Transpose.hpp @@ -0,0 +1,21 @@ +// +// Copyright © 2020 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include <armnn/TensorFwd.hpp> +#include <armnn/Types.hpp> + +namespace armnnUtils +{ + +armnn::TensorShape TransposeTensorShape(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings); + +armnn::TensorInfo TransposeTensorShape(const armnn::TensorInfo& info, const armnn::PermutationVector& mappings); + +void Transpose(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings, + const void* src, void* dst, size_t dataTypeSize); + +} // namespace armnnUtils |