aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/DetectionPostProcessLayer.hpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-01-25 10:46:40 +0000
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-01-25 16:59:16 +0000
commita0d56c77a53f5f991565041927003ce7460730ce (patch)
treede08bfa6f95752ec8e7b5b58d0b689288f44b36a /src/armnn/layers/DetectionPostProcessLayer.hpp
parent12055747d47657a89d60748a078897f6436e6aa0 (diff)
downloadarmnn-a0d56c77a53f5f991565041927003ce7460730ce.tar.gz
IVGCVSW-2556 Add Layer implementation for Detection PostProcess
* Add DetectionPostProcessDescriptor. * Add implementation for DetectionPostProcessLayer. * Unit test to validate output. Change-Id: If63e83eb2a2978c549071c7aeb272906e7c35fe9
Diffstat (limited to 'src/armnn/layers/DetectionPostProcessLayer.hpp')
-rw-r--r--src/armnn/layers/DetectionPostProcessLayer.hpp9
1 files changed, 9 insertions, 0 deletions
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