diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-16 14:27:52 +0000 |
|---|---|---|
| committer | Jim Flynn <jim.flynn@arm.com> | 2020-11-17 12:21:42 +0000 |
| commit | aa41d5d2f43790938f3a32586626be5ef55b6ca9 (patch) | |
| tree | 9b6daa8b9b80b7c6d939c894d1ae3715a565bdac /src/backends/reference/RefLayerSupport.cpp | |
| parent | 7c0e3fd7596abde1cf2eca96147e1c37491e1816 (diff) | |
| download | armnn-aa41d5d2f43790938f3a32586626be5ef55b6ca9.tar.gz | |
IVGCVSW-5530 'Cannot run SSD Mobilenet f16/uint8 on CpuRef via ExecuteNetwork'
* Added FP16 DataType support to DetectionPostProcess
* For DetectionPostProcess layer output is always Float32 regardless of input type
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I21f63dd08f0863e9a98e105b3009bab3da1ab0c3
Diffstat (limited to 'src/backends/reference/RefLayerSupport.cpp')
| -rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index f48c120203..b3feae6713 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -722,10 +722,11 @@ bool RefLayerSupport::IsDetectionPostProcessSupported(const TensorInfo& boxEncod bool supported = true; - std::array<DataType,5> supportedInputTypes = + std::array<DataType,6> supportedInputTypes = { DataType::BFloat16, DataType::Float32, + DataType::Float16, DataType::QAsymmS8, DataType::QAsymmU8, DataType::QSymmS16 |