diff options
| author | nikraj01 <nikhil.raj@arm.com> | 2019-06-14 14:20:40 +0100 |
|---|---|---|
| committer | nikraj01 <nikhil.raj@arm.com> | 2019-06-14 14:20:40 +0100 |
| commit | 24d7321ad7897e8836d4a38039a73a0ec419cf43 (patch) | |
| tree | 9401dcc97d4843e66473b8fbf8d07c40c561750e /src/backends/reference/RefLayerSupport.cpp | |
| parent | c6138d8a8af334fad5230d73e456f303f9665bae (diff) | |
| download | armnn-24d7321ad7897e8836d4a38039a73a0ec419cf43.tar.gz | |
IVGCVSW-3225 Add QSymm16 support for Rsqrt workload
Change-Id: I83b8494af24ff271dc4cd609944a1c5c55c405e0
Signed-off-by: nikraj01 <nikhil.raj@arm.com>
Diffstat (limited to 'src/backends/reference/RefLayerSupport.cpp')
| -rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index d8c942cd96..03c8633dce 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -1136,10 +1136,11 @@ bool RefLayerSupport::IsRsqrtSupported(const TensorInfo& input, Optional<std::string&> reasonIfUnsupported) const { bool supported = true; - std::array<DataType,2> supportedTypes = + std::array<DataType,3> supportedTypes = { DataType::Float32, - DataType::QuantisedAsymm8 + DataType::QuantisedAsymm8, + DataType::QuantisedSymm16 }; supported &= CheckSupportRule(TypeAnyOf(input, supportedTypes), reasonIfUnsupported, |