From dea2d2d58fe3a742e6f66fe50befbe0044e15ad1 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 19 Dec 2018 16:23:17 +0000 Subject: COMPMID-1772: Implement PadV2 for NEON Change-Id: Ia4604524a034c46b004fd850183480c5fbfd8cb3 Reviewed-on: https://review.mlplatform.org/437 Tested-by: Arm Jenkins Reviewed-by: Michalis Spyrou --- src/runtime/CL/functions/CLPadLayer.cpp | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'src/runtime/CL/functions/CLPadLayer.cpp') diff --git a/src/runtime/CL/functions/CLPadLayer.cpp b/src/runtime/CL/functions/CLPadLayer.cpp index de43c7dca2..3aa1b1e1a0 100644 --- a/src/runtime/CL/functions/CLPadLayer.cpp +++ b/src/runtime/CL/functions/CLPadLayer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018 ARM Limited. + * Copyright (c) 2018-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -34,21 +34,21 @@ CLPadLayer::CLPadLayer() { } -void CLPadLayer::configure(ICLTensor *input, ICLTensor *output, const PaddingList &padding) +void CLPadLayer::configure(ICLTensor *input, ICLTensor *output, const PaddingList &padding, PixelValue constant_value) { // Copy the input to the output _copy_kernel.configure(input, output, padding); // Set the pages of the output to zero - _memset_kernel.configure(output, PixelValue()); + _memset_kernel.configure(output, constant_value); // Fill padding on the first two dimensions with zeros - _fillborder_kernel.configure(input, input->info()->padding(), BorderMode::CONSTANT); + _fillborder_kernel.configure(input, input->info()->padding(), BorderMode::CONSTANT, constant_value); } -Status CLPadLayer::validate(const ITensorInfo *input, const ITensorInfo *output, const PaddingList &padding) +Status CLPadLayer::validate(const ITensorInfo *input, const ITensorInfo *output, const PaddingList &padding, PixelValue constant_value) { - ARM_COMPUTE_RETURN_ON_ERROR(CLMemsetKernel::validate(input, PixelValue())); + ARM_COMPUTE_RETURN_ON_ERROR(CLMemsetKernel::validate(input, constant_value)); ARM_COMPUTE_RETURN_ON_ERROR(CLCopyKernel::validate(input, output, padding)); return Status{}; -- cgit v1.2.1