diff options
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/Types.hpp | 2 | ||||
| -rw-r--r-- | include/armnn/TypesUtils.hpp | 4 |
2 files changed, 4 insertions, 2 deletions
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index df5e6a7199..dcc8c9e52c 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -37,7 +37,7 @@ enum class DataType Boolean = 4, QuantisedSymm16 = 5, QuantizedSymm8PerAxis = 6, - QuantisedSymm8 = 7 + QSymmS8 = 7 }; enum class DataLayout 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; } |