diff options
Diffstat (limited to 'include/armnn/TypesUtils.hpp')
| -rw-r--r-- | include/armnn/TypesUtils.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index c8f612f08d..4394d9ddd9 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -116,6 +116,7 @@ constexpr unsigned int GetDataTypeSize(DataType dataType) case DataType::Float32: case DataType::Signed32: return 4U; case DataType::QuantisedAsymm8: return 1U; + case DataType::QSymmS8: return 1U; case DataType::QuantizedSymm8PerAxis: return 1U; case DataType::QuantisedSymm16: return 2U; case DataType::Boolean: return 1U; @@ -163,6 +164,7 @@ constexpr const char* GetDataTypeName(DataType dataType) case DataType::Float16: return "Float16"; case DataType::Float32: return "Float32"; case DataType::QuantisedAsymm8: return "QAsymm8"; + case DataType::QSymmS8: return "QSymm8"; case DataType::QuantizedSymm8PerAxis: return "QSymm8PerAxis"; case DataType::QuantisedSymm16: return "QSymm16"; case DataType::Signed32: return "Signed32"; @@ -198,7 +200,7 @@ constexpr bool IsQuantizedType() constexpr bool IsQuantizedType(DataType dataType) { return dataType == DataType::QuantisedAsymm8 || - dataType == DataType::QuantisedSymm8 || + dataType == DataType::QSymmS8 || dataType == DataType::QuantisedSymm16 || dataType == DataType::QuantizedSymm8PerAxis; } |