diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/TypesUtils.hpp | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 59beb33144..bf54c15ef8 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -169,6 +169,7 @@ constexpr const char* GetDataTypeName(DataType dataType) case DataType::Float16: return "Float16"; case DataType::Float32: return "Float32"; case DataType::QAsymmU8: return "QAsymmU8"; + case DataType::QAsymmS8: return "QAsymmS8"; case DataType::QSymmS8: return "QSymmS8"; ARMNN_NO_DEPRECATE_WARN_BEGIN case DataType::QuantizedSymm8PerAxis: return "QSymm8PerAxis"; @@ -233,17 +234,21 @@ constexpr bool IsQuantizedType() return std::is_integral<T>::value; } -constexpr bool IsQuantizedType(DataType dataType) +constexpr bool IsQuantized8BitType(DataType dataType) { ARMNN_NO_DEPRECATE_WARN_BEGIN return dataType == DataType::QAsymmU8 || dataType == DataType::QAsymmS8 || dataType == DataType::QSymmS8 || - dataType == DataType::QSymmS16 || dataType == DataType::QuantizedSymm8PerAxis; ARMNN_NO_DEPRECATE_WARN_END } +constexpr bool IsQuantizedType(DataType dataType) +{ + return dataType == DataType::QSymmS16 || IsQuantized8BitType(dataType); +} + inline std::ostream& operator<<(std::ostream& os, Status stat) { os << GetStatusAsCString(stat); |