diff options
Diffstat (limited to 'src/armnn/CompatibleTypes.hpp')
| -rw-r--r-- | src/armnn/CompatibleTypes.hpp | 6 |
1 files changed, 6 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; |