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 4e31aed2aa..06aa064f7b 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<int8_t>(DataType dataType) +{ + return dataType == DataType::QuantisedSymm8 || dataType == DataType::QuantizedSymm8PerAxis; +} + +template<> inline bool CompatibleTypes<int16_t>(DataType dataType) { return dataType == DataType::QuantisedSymm16; |