diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-11-26 12:50:34 +0000 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-11-27 09:53:32 +0000 |
commit | e9323ec3d8c638462e7269ecbef7d845b6a8b11a (patch) | |
tree | 930d60a90fe65ca44cab2456105e464c2f1abcf7 /include/armnn/TypesUtils.hpp | |
parent | 94c4fef71594a60e4ea27aa6fa794c7ed3be4a27 (diff) | |
download | armnn-e9323ec3d8c638462e7269ecbef7d845b6a8b11a.tar.gz |
Add IsQuantizeType(DataType) for use in validation functions
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Ib790b24cfde87c92e29615da942c25bf7c98f95e
Diffstat (limited to 'include/armnn/TypesUtils.hpp')
-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); |