aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/DetectionPostProcessLayer.cpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-01-23 18:06:26 +0000
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-01-24 15:02:11 +0000
commit94dd5d87a1d3a05a2289e309a77a3851d9ec2741 (patch)
tree4661e8f4a792e2468daff4157e4a9f089fc06da0 /src/armnn/layers/DetectionPostProcessLayer.cpp
parent2b4d88e34ac1f965417fd236fd4786f26bae2042 (diff)
downloadarmnn-94dd5d87a1d3a05a2289e309a77a3851d9ec2741.tar.gz
IVGCVSW-2555 Add no-op implementation for Detection PostProcess
* Added DetectionPostProcessQueueDescriptor to WorkloadData * Added CreateDetectionPostProcess function in WorkloadFactory.hpp * Added stub implementation of the CreateDetectionPostProcess in workload factories * Added DetectionPostProcessLayer stub implementation * Added AddDetectionPostProcessLayer to Network * Added IsDetectionPostProcessSupported to LayerSupportBase Change-Id: Ifc071b3b6b12877c997bdcc43d769c8f891d5c6c
Diffstat (limited to 'src/armnn/layers/DetectionPostProcessLayer.cpp')
-rw-r--r--src/armnn/layers/DetectionPostProcessLayer.cpp39
1 files changed, 39 insertions, 0 deletions
diff --git a/src/armnn/layers/DetectionPostProcessLayer.cpp b/src/armnn/layers/DetectionPostProcessLayer.cpp
new file mode 100644
index 0000000000..99aaac7b9f
--- /dev/null
+++ b/src/armnn/layers/DetectionPostProcessLayer.cpp
@@ -0,0 +1,39 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "DetectionPostProcessLayer.hpp"
+
+#include "LayerCloneBase.hpp"
+
+#include <armnn/TypesUtils.hpp>
+#include <backendsCommon/WorkloadData.hpp>
+#include <backendsCommon/WorkloadFactory.hpp>
+
+namespace armnn
+{
+
+DetectionPostProcessLayer::DetectionPostProcessLayer(const DetectionPostProcessDescriptor& param, const char* name)
+ : LayerWithParameters(2, 4, LayerType::DetectionPostProcess, param, name)
+{
+}
+
+std::unique_ptr<IWorkload> DetectionPostProcessLayer::CreateWorkload(const armnn::Graph& graph,
+ const armnn::IWorkloadFactory& factory) const
+{
+ DetectionPostProcessQueueDescriptor descriptor;
+ return factory.CreateDetectionPostProcess(descriptor, PrepInfoAndDesc(descriptor, graph));
+}
+
+DetectionPostProcessLayer* DetectionPostProcessLayer::Clone(Graph& graph) const
+{
+ return CloneBase<DetectionPostProcessLayer>(graph, m_Param, GetName());
+}
+
+void DetectionPostProcessLayer::ValidateTensorShapesFromInputs()
+{
+ VerifyLayerConnections(2, CHECK_LOCATION());
+}
+
+} // namespace armnn