diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-02-04 11:46:26 +0000 |
|---|---|---|
| committer | Nina Drozd <nina.drozd@arm.com> | 2019-02-08 10:07:20 +0000 |
| commit | 6d302bfb568962f3b6b6f012b260ce54f22d36a0 (patch) | |
| tree | 86147d527e36751392539f363e5aa702e49f2fc7 /src/backends/reference/RefWorkloadFactory.cpp | |
| parent | 61980d472006abdf3778d23903fb3bec5916f1f2 (diff) | |
| download | armnn-6d302bfb568962f3b6b6f012b260ce54f22d36a0.tar.gz | |
IVGCVSW-2559 End to end tests for Detection PostProcess
* end to end tests for Detection PostProcess float and uint8
* add anchors to AddDetectionPostProcessLayer
* add anchors to VisitDetectionPostProcessLayer
* refactor code
Change-Id: I3c5a9a4a60b74c2246b4a27692bbf3c235163f90
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Diffstat (limited to 'src/backends/reference/RefWorkloadFactory.cpp')
| -rw-r--r-- | src/backends/reference/RefWorkloadFactory.cpp | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp index 9c1ce1e013..3bf83bd9be 100644 --- a/src/backends/reference/RefWorkloadFactory.cpp +++ b/src/backends/reference/RefWorkloadFactory.cpp @@ -154,7 +154,16 @@ std::unique_ptr<IWorkload> RefWorkloadFactory::CreateDepthwiseConvolution2d( std::unique_ptr<IWorkload> RefWorkloadFactory::CreateDetectionPostProcess( const armnn::DetectionPostProcessQueueDescriptor& descriptor, const armnn::WorkloadInfo& info) const { - return MakeWorkload<RefDetectionPostProcessFloat32Workload, RefDetectionPostProcessUint8Workload>(descriptor, info); + const DataType dataType = info.m_InputTensorInfos[0].GetDataType(); + switch (dataType) + { + case DataType::Float32: + return std::make_unique<RefDetectionPostProcessFloat32Workload>(descriptor, info); + case DataType::QuantisedAsymm8: + return std::make_unique<RefDetectionPostProcessUint8Workload>(descriptor, info); + default: + return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info); + } } std::unique_ptr<armnn::IWorkload> RefWorkloadFactory::CreateNormalization( |