commit 07fbe3707608bf0c08efb42e15d9b1afaf16ef1e
author Georgios Pinitas <georgios.pinitas@arm.com> Tue Feb 04 13:21:02 2020 +0000
committer Georgios Pinitas <georgios.pinitas@arm.com> Wed Feb 05 13:44:15 2020 +0000
tree 0d2edea779a5f44c037de53ef120722ddb273167
parent 5c829cab49b0dcffd893065af05f0bf028e7a548

COMPMID-2819: Enable PadLayer on NEON graph backend.

Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I0114d6999413cbb0602db19c68b9c41c419c6df0
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2682
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>

arm_compute/graph/nodes/PadLayerNode.h

diff --git a/arm_compute/graph/nodes/PadLayerNode.h b/arm_compute/graph/nodes/PadLayerNode.h
index 7d40ae8..0b1bdda 100644
--- a/arm_compute/graph/nodes/PadLayerNode.h
+++ b/arm_compute/graph/nodes/PadLayerNode.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2018-2019 ARM Limited.
+ * Copyright (c) 2018-2020 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -39,7 +39,7 @@
      * @param[in] padding The padding for each spatial dimension of the input tensor. The pair padding[i]
      *                    specifies the front and the end padding in the i-th dimension.
      */
-    PadLayerNode(PaddingList &padding);
+    PadLayerNode(const PaddingList &padding);
     /** Padding list accessor
      *
      * @return Padding list

src/graph/backends/NEON/NEFunctionFactory.cpp

diff --git a/src/graph/backends/NEON/NEFunctionFactory.cpp b/src/graph/backends/NEON/NEFunctionFactory.cpp
index d72cffa..3597139 100644
--- a/src/graph/backends/NEON/NEFunctionFactory.cpp
+++ b/src/graph/backends/NEON/NEFunctionFactory.cpp
@@ -225,6 +225,8 @@
             return detail::create_fused_depthwise_convolution_batch_normalization_layer<NEFusedLayerTypes, NETargetInfo>(*polymorphic_downcast<FusedDepthwiseConvolutionBatchNormalizationNode *>(node), ctx);
         case NodeType::NormalizationLayer:
             return detail::create_normalization_layer<NENormalizationLayer, NETargetInfo>(*polymorphic_downcast<NormalizationLayerNode *>(node), ctx);
+        case NodeType::PadLayer:
+            return detail::create_pad_layer<NEPadLayer, NETargetInfo>(*polymorphic_downcast<PadLayerNode *>(node));
         case NodeType::PermuteLayer:
             return detail::create_permute_layer<NEPermute, NETargetInfo>(*polymorphic_downcast<PermuteLayerNode *>(node));
         case NodeType::PoolingLayer:

src/graph/backends/NEON/NENodeValidator.cpp

diff --git a/src/graph/backends/NEON/NENodeValidator.cpp b/src/graph/backends/NEON/NENodeValidator.cpp
index fc84959..ec8c0e5 100644
--- a/src/graph/backends/NEON/NENodeValidator.cpp
+++ b/src/graph/backends/NEON/NENodeValidator.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2018-2019 ARM Limited.
+ * Copyright (c) 2018-2020 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -70,7 +70,7 @@
         case NodeType::NormalizePlanarYUVLayer:
             return ARM_COMPUTE_CREATE_ERROR(arm_compute::ErrorCode::RUNTIME_ERROR, "Unsupported operation : NormalizePlanarYUVLayer");
         case NodeType::PadLayer:
-            return ARM_COMPUTE_CREATE_ERROR(arm_compute::ErrorCode::RUNTIME_ERROR, "Unsupported operation : PadLayer");
+            return detail::validate_pad_layer<NEPadLayer>(*polymorphic_downcast<PadLayerNode *>(node));
         case NodeType::PermuteLayer:
             return detail::validate_permute_layer<NEPermute>(*polymorphic_downcast<PermuteLayerNode *>(node));
         case NodeType::PriorBoxLayer:

src/graph/nodes/PadLayerNode.cpp

diff --git a/src/graph/nodes/PadLayerNode.cpp b/src/graph/nodes/PadLayerNode.cpp
index e7996d2..1bd0776 100644
--- a/src/graph/nodes/PadLayerNode.cpp
+++ b/src/graph/nodes/PadLayerNode.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2018 ARM Limited.
+ * Copyright (c) 2018-2020 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -32,7 +32,7 @@
 {
 namespace graph
 {
-PadLayerNode::PadLayerNode(PaddingList &padding)
+PadLayerNode::PadLayerNode(const PaddingList &padding)
     : _padding(padding)
 {
     _input_edges.resize(1, EmptyEdgeID);