aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/TypesUtils.hpp
diff options
context:
space:
mode:
authorKeith Davis <keith.davis@arm.com>2020-02-11 16:51:50 +0000
committerJames Conroy <james.conroy@arm.com>2020-02-17 21:53:29 +0000
commit0c2eeac6347533a1d3d456aebea492f5123388f3 (patch)
treef218fc236137791c491b680dfd24fb9706c171a6 /include/armnn/TypesUtils.hpp
parent4c3c1f486ab775eacb1f6455f8468f9be2c3e4f7 (diff)
downloadarmnn-0c2eeac6347533a1d3d456aebea492f5123388f3.tar.gz
IVGCVSW-4436 Add ExecuteNetwork test for mobilenet_v2_int8
* Add QAsymmS8 to QueueDescriptor supportedTypes * Add QSymmS8/QAsymmS8 to RefLayerSupport supportedTypes * Some additional comments and refactoring Change-Id: I8567314452e6e8f6f69cb6e458ee147d3fc92fab Signed-off-by: Keith Davis <keith.davis@arm.com>
Diffstat (limited to 'include/armnn/TypesUtils.hpp')
-rw-r--r--include/armnn/TypesUtils.hpp9
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);