diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-19 12:26:36 +0000 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-19 12:26:36 +0000 |
commit | cd5ac23039dc3acd750c44442960280cc33307ef (patch) | |
tree | 37de0bae199345355535448b6c1ce2c624a298b8 /src/armnn | |
parent | 2a326b5bbff91ffaedd9017f388aa45862047743 (diff) | |
download | armnn-cd5ac23039dc3acd750c44442960280cc33307ef.tar.gz |
IVGCVSW-2839 Add QuantisedSymm16 support to the ArmNN frontend
Change-Id: I76f4e7db55b0efbb1a5f7d1a32bf451e66add0c7
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/CompatibleTypes.hpp | 6 | ||||
-rw-r--r-- | src/armnn/TypeUtils.hpp | 6 |
2 files changed, 12 insertions, 0 deletions
diff --git a/src/armnn/CompatibleTypes.hpp b/src/armnn/CompatibleTypes.hpp index 2449876544..4e31aed2aa 100644 --- a/src/armnn/CompatibleTypes.hpp +++ b/src/armnn/CompatibleTypes.hpp @@ -36,6 +36,12 @@ inline bool CompatibleTypes<uint8_t>(DataType dataType) } template<> +inline bool CompatibleTypes<int16_t>(DataType dataType) +{ + return dataType == DataType::QuantisedSymm16; +} + +template<> inline bool CompatibleTypes<int32_t>(DataType dataType) { return dataType == DataType::Signed32; diff --git a/src/armnn/TypeUtils.hpp b/src/armnn/TypeUtils.hpp index f7d0e077c8..55269f4620 100644 --- a/src/armnn/TypeUtils.hpp +++ b/src/armnn/TypeUtils.hpp @@ -33,6 +33,12 @@ struct ResolveTypeImpl<DataType::QuantisedAsymm8> }; template<> +struct ResolveTypeImpl<DataType::QuantisedSymm16> +{ + using Type = int16_t; +}; + +template<> struct ResolveTypeImpl<DataType::Signed32> { using Type = int32_t; |