commit 8cf8c1123440c2002ee108d1949529bf21eac944
author Usama Arif <usama.arif@arm.com> Thu Mar 14 15:36:54 2019 +0000
committer Pablo Marquez <pablo.tello@arm.com> Wed Mar 27 09:22:04 2019 +0000
tree cc61d9ed5ee805c4356b8497b2e81f67b194b36a
parent adc2186c06ca27f368dfe6ceadce449551259efc

COMPMID-1944 Add support for "reflect" padding mode in NEPad

Change-Id: I56c42524497d37d44708648571fa211ac1afbd98
Signed-off-by: Usama Arif <usama.arif@arm.com>
Reviewed-on: https://review.mlplatform.org/c/885
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>

src/runtime/CL/functions/CLPadLayer.cpp

diff --git a/src/runtime/CL/functions/CLPadLayer.cpp b/src/runtime/CL/functions/CLPadLayer.cpp
index 3aa1b1e..fac2364 100644
--- a/src/runtime/CL/functions/CLPadLayer.cpp
+++ b/src/runtime/CL/functions/CLPadLayer.cpp
@@ -34,8 +34,9 @@
 {
 }
 
-void CLPadLayer::configure(ICLTensor *input, ICLTensor *output, const PaddingList &padding, PixelValue constant_value)
+void CLPadLayer::configure(ICLTensor *input, ICLTensor *output, const PaddingList &padding, PixelValue constant_value, PaddingMode mode)
 {
+    ARM_COMPUTE_UNUSED(mode);
     // Copy the input to the output
     _copy_kernel.configure(input, output, padding);
 
@@ -46,10 +47,11 @@
     _fillborder_kernel.configure(input, input->info()->padding(), BorderMode::CONSTANT, constant_value);
 }
 
-Status CLPadLayer::validate(const ITensorInfo *input, const ITensorInfo *output, const PaddingList &padding, PixelValue constant_value)
+Status CLPadLayer::validate(const ITensorInfo *input, const ITensorInfo *output, const PaddingList &padding, PixelValue constant_value, PaddingMode mode)
 {
     ARM_COMPUTE_RETURN_ON_ERROR(CLMemsetKernel::validate(input, constant_value));
     ARM_COMPUTE_RETURN_ON_ERROR(CLCopyKernel::validate(input, output, padding));
+    ARM_COMPUTE_RETURN_ERROR_ON(mode != PaddingMode::CONSTANT);
 
     return Status{};
 }