diff options
Diffstat (limited to 'src/armnn/backends/RefWorkloads/RefWorkloadUtils.hpp')
-rw-r--r-- | src/armnn/backends/RefWorkloads/RefWorkloadUtils.hpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/armnn/backends/RefWorkloads/RefWorkloadUtils.hpp b/src/armnn/backends/RefWorkloads/RefWorkloadUtils.hpp index 088fe819e5..1df735ea55 100644 --- a/src/armnn/backends/RefWorkloads/RefWorkloadUtils.hpp +++ b/src/armnn/backends/RefWorkloads/RefWorkloadUtils.hpp @@ -9,6 +9,7 @@ #include <armnn/Tensor.hpp> #include <armnn/Types.hpp> +#include <Half.hpp> #include <boost/polymorphic_cast.hpp> @@ -70,6 +71,18 @@ float* GetOutputTensorDataFloat(unsigned int idx, const PayloadType& data) return GetOutputTensorData<float>(idx, data); } +template <typename PayloadType> +const Half* GetInputTensorDataHalf(unsigned int idx, const PayloadType& data) +{ + return GetInputTensorData<Half>(idx, data); +} + +template <typename PayloadType> +Half* GetOutputTensorDataHalf(unsigned int idx, const PayloadType& data) +{ + return GetOutputTensorData<Half>(idx, data); +} + //////////////////////////////////////////// /// u8 helpers //////////////////////////////////////////// |