diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2019-12-04 14:27:27 +0000 |
|---|---|---|
| committer | Jim Flynn Arm <jim.flynn@arm.com> | 2019-12-09 15:39:16 +0000 |
| commit | fd2710651ada27fc82f28c07fb1e09effc3bda2d (patch) | |
| tree | 7c2200489c7a3f845b91362c2c8d66ab9c6101e8 /src/backends/reference/RefLayerSupport.cpp | |
| parent | 6a5e5e8b7e56f927d70ced3203d6e16df3fdd189 (diff) | |
| download | armnn-fd2710651ada27fc82f28c07fb1e09effc3bda2d.tar.gz | |
IVGCVSW-4211 Add Signed 8 bit Quantisation support into the Reference backend
!android-nn-driver:2435
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I10ecd4a8937725953396805f33a3562a5384c4d4
Diffstat (limited to 'src/backends/reference/RefLayerSupport.cpp')
| -rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 299503ddc6..19b76152f3 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -615,8 +615,9 @@ bool RefLayerSupport::IsDequantizeSupported(const TensorInfo& input, { bool supported = true; - std::array<DataType,2> supportedInputTypes = { + std::array<DataType,3> supportedInputTypes = { DataType::QuantisedAsymm8, + DataType::QSymmS8, DataType::QuantisedSymm16 }; @@ -1398,7 +1399,7 @@ bool RefLayerSupport::IsQuantizeSupported(const TensorInfo& input, { bool supported = true; - // Define supported output types. + // Define supported input types. std::array<DataType,1> supportedInputTypes = { DataType::Float32, }; @@ -1407,8 +1408,9 @@ bool RefLayerSupport::IsQuantizeSupported(const TensorInfo& input, "Reference quantize: input type not supported."); // Define supported output types. - std::array<DataType,2> supportedOutputTypes = { + std::array<DataType,3> supportedOutputTypes = { DataType::QuantisedAsymm8, + DataType::QSymmS8, DataType::QuantisedSymm16 }; supported &= CheckSupportRule(TypeAnyOf(output, supportedOutputTypes), reasonIfUnsupported, |