commit 6a5e5e8b7e56f927d70ced3203d6e16df3fdd189
author Derek Lamberti <derek.lamberti@arm.com> Thu Dec 05 14:41:20 2019 +0000
committer Derek Lamberti <derek.lamberti@arm.com> Mon Dec 09 14:22:15 2019 +0000
tree d961fa1349c6a51a595e51c29c1df42419a7a426
parent e5b8eb9fe8147a0849db08ef0898a0e8bef920b4

IVGCVSW-4227 Add CpuAcc backend support for DetectionPostProcess

Change-Id: I318bf92b8d1db593d9c30b9b4412bfecbe65bc12
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>

src/backends/reference/RefLayerSupport.cpp

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 30ce5c6..299503d 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -637,10 +637,15 @@
     return supported;
 }
 
-bool RefLayerSupport::IsDetectionPostProcessSupported(const armnn::TensorInfo& input0,
-                                                      const armnn::TensorInfo& input1,
-                                                      const armnn::DetectionPostProcessDescriptor& descriptor,
-                                                      armnn::Optional<std::string&> reasonIfUnsupported) const
+bool RefLayerSupport::IsDetectionPostProcessSupported(const TensorInfo& boxEncodings,
+                                                      const TensorInfo& scores,
+                                                      const TensorInfo& anchors,
+                                                      const TensorInfo& detectionBoxes,
+                                                      const TensorInfo& detectionClasses,
+                                                      const TensorInfo& detectionScores,
+                                                      const TensorInfo& numDetections,
+                                                      const DetectionPostProcessDescriptor& descriptor,
+                                                      Optional<std::string&> reasonIfUnsupported) const
 {
     bool supported = true;
 
@@ -651,10 +656,10 @@
         DataType::QuantisedSymm16
     };
 
-    supported &= CheckSupportRule(TypeAnyOf(input0, supportedInputTypes), reasonIfUnsupported,
+    supported &= CheckSupportRule(TypeAnyOf(boxEncodings, supportedInputTypes), reasonIfUnsupported,
                                   "Reference DetectionPostProcess: input 0 is not a supported type.");
 
-    supported &= CheckSupportRule(TypeAnyOf(input1, supportedInputTypes), reasonIfUnsupported,
+    supported &= CheckSupportRule(TypeAnyOf(scores, supportedInputTypes), reasonIfUnsupported,
                                   "Reference DetectionPostProcess: input 1 is not a supported type.");
 
     return supported;