From a0d56c77a53f5f991565041927003ce7460730ce Mon Sep 17 00:00:00 2001
From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Fri, 25 Jan 2019 10:46:40 +0000
Subject: IVGCVSW-2556 Add Layer implementation for Detection PostProcess

 * Add DetectionPostProcessDescriptor.
 * Add implementation for DetectionPostProcessLayer.
 * Unit test to validate output.

Change-Id: If63e83eb2a2978c549071c7aeb272906e7c35fe9
---
 src/armnn/layers/DetectionPostProcessLayer.hpp | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src/armnn/layers/DetectionPostProcessLayer.hpp')

diff --git a/src/armnn/layers/DetectionPostProcessLayer.hpp b/src/armnn/layers/DetectionPostProcessLayer.hpp
index 629e3864b3..a1c499e793 100644
--- a/src/armnn/layers/DetectionPostProcessLayer.hpp
+++ b/src/armnn/layers/DetectionPostProcessLayer.hpp
@@ -10,10 +10,15 @@
 namespace armnn
 {
 
+class ScopedCpuTensorHandle;
+
 /// This layer represents a detection postprocess operator.
 class DetectionPostProcessLayer : public LayerWithParameters<DetectionPostProcessDescriptor>
 {
 public:
+    /// A unique pointer to store Anchor values.
+    std::unique_ptr<ScopedCpuTensorHandle> m_Anchors;
+
     /// Makes a workload for the DetectionPostProcess type.
     /// @param [in] graph The graph where this layer can be found.
     /// @param [in] factory The workload factory which will create the workload.
@@ -39,6 +44,10 @@ protected:
 
     /// Default destructor
     ~DetectionPostProcessLayer() = default;
+
+    /// Retrieve the handles to the constant values stored by the layer.
+    /// @return A vector of the constant tensors stored by this layer.
+    ConstantTensors GetConstantTensorsByRef() override;
 };
 
 } // namespace armnn
-- 
cgit v1.2.1