diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-11-26 14:04:54 +0000 |
---|---|---|
committer | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-11-26 15:00:28 +0000 |
commit | 21fc28bb68d7a6232eea3a0e8cfd800d95346ea4 (patch) | |
tree | 5eca96ac31e3f8c62da3ad7742834180a0f7eeac /src/armnn | |
parent | 9b6d4246804d39e5b7570992d63e31a044442385 (diff) | |
download | armnn-21fc28bb68d7a6232eea3a0e8cfd800d95346ea4.tar.gz |
IVGCVSW-3866 Add support for per-channel QSymm8 Convolution2d to Neon backend
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: If7d366b2894050c53a651016165f34be1a2ce7a6
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/CompatibleTypes.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/CompatibleTypes.hpp b/src/armnn/CompatibleTypes.hpp index 4e31aed2aa..06aa064f7b 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<int8_t>(DataType dataType) +{ + return dataType == DataType::QuantisedSymm8 || dataType == DataType::QuantizedSymm8PerAxis; +} + +template<> inline bool CompatibleTypes<int16_t>(DataType dataType) { return dataType == DataType::QuantisedSymm16; |