diff options
Diffstat (limited to 'src/armnnUtils/Permute.hpp')
| -rw-r--r-- | src/armnnUtils/Permute.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnnUtils/Permute.hpp b/src/armnnUtils/Permute.hpp index 700ddc72ce..4e4319822b 100644 --- a/src/armnnUtils/Permute.hpp +++ b/src/armnnUtils/Permute.hpp @@ -14,7 +14,10 @@ armnn::TensorShape Permuted(const armnn::TensorShape& srcShape, const armnn::Per 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); + template <typename T> void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings, const T* src, T* dst); -} // namespace armnnUtils
\ No newline at end of file +} // namespace armnnUtils |