diff options
Diffstat (limited to 'src/armnn/TypeUtils.hpp')
-rw-r--r-- | src/armnn/TypeUtils.hpp | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/src/armnn/TypeUtils.hpp b/src/armnn/TypeUtils.hpp index 01a0e6479a..5bb040f780 100644 --- a/src/armnn/TypeUtils.hpp +++ b/src/armnn/TypeUtils.hpp @@ -11,16 +11,9 @@ namespace armnn { - template<DataType DT> struct ResolveTypeImpl; -template<> -struct ResolveTypeImpl<DataType::QuantisedAsymm8> -{ - using Type = uint8_t; -}; - template <> struct ResolveTypeImpl<DataType::Float16> { @@ -34,6 +27,18 @@ struct ResolveTypeImpl<DataType::Float32> }; template<> +struct ResolveTypeImpl<DataType::QuantisedAsymm8> +{ + using Type = uint8_t; +}; + +template<> +struct ResolveTypeImpl<DataType::Signed32> +{ + using Type = int32_t; +}; + +template<> struct ResolveTypeImpl<DataType::Boolean> { using Type = bool; @@ -42,5 +47,4 @@ struct ResolveTypeImpl<DataType::Boolean> template<DataType DT> using ResolveType = typename ResolveTypeImpl<DT>::Type; - } //namespace armnn |