diff options
Diffstat (limited to 'include/armnn')
-rw-r--r-- | include/armnn/TypesUtils.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 32967c6b42..c8f612f08d 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -195,6 +195,14 @@ constexpr bool IsQuantizedType() return std::is_integral<T>::value; } +constexpr bool IsQuantizedType(DataType dataType) +{ + return dataType == DataType::QuantisedAsymm8 || + dataType == DataType::QuantisedSymm8 || + dataType == DataType::QuantisedSymm16 || + dataType == DataType::QuantizedSymm8PerAxis; +} + inline std::ostream& operator<<(std::ostream& os, Status stat) { os << GetStatusAsCString(stat); |