diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/CompatibleTypes.hpp | 6 | ||||
-rw-r--r-- | src/armnn/TypeUtils.hpp | 6 |
2 files changed, 12 insertions, 0 deletions
diff --git a/src/armnn/CompatibleTypes.hpp b/src/armnn/CompatibleTypes.hpp index 2449876544..4e31aed2aa 100644 --- a/src/armnn/CompatibleTypes.hpp +++ b/src/armnn/CompatibleTypes.hpp @@ -36,6 +36,12 @@ inline bool CompatibleTypes<uint8_t>(DataType dataType) } template<> +inline bool CompatibleTypes<int16_t>(DataType dataType) +{ + return dataType == DataType::QuantisedSymm16; +} + +template<> inline bool CompatibleTypes<int32_t>(DataType dataType) { return dataType == DataType::Signed32; diff --git a/src/armnn/TypeUtils.hpp b/src/armnn/TypeUtils.hpp index f7d0e077c8..55269f4620 100644 --- a/src/armnn/TypeUtils.hpp +++ b/src/armnn/TypeUtils.hpp @@ -33,6 +33,12 @@ struct ResolveTypeImpl<DataType::QuantisedAsymm8> }; template<> +struct ResolveTypeImpl<DataType::QuantisedSymm16> +{ + using Type = int16_t; +}; + +template<> struct ResolveTypeImpl<DataType::Signed32> { using Type = int32_t; |