diff options
Diffstat (limited to 'src/armnnUtils/Permute.cpp')
| -rw-r--r-- | src/armnnUtils/Permute.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnnUtils/Permute.cpp b/src/armnnUtils/Permute.cpp index 9fe198b3a7..61f4e0e644 100644 --- a/src/armnnUtils/Permute.cpp +++ b/src/armnnUtils/Permute.cpp @@ -5,6 +5,7 @@ #include "Permute.hpp" +#include "Half.hpp" #include <armnn/Tensor.hpp> #include <cassert> @@ -109,6 +110,8 @@ void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& // Instantiates for types. template void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings, + const armnn::Half* src, armnn::Half* dst); +template void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings, const float* src, float* dst); template void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings, const uint8_t* src, uint8_t* dst); |