diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-12-19 16:23:17 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-01-14 16:26:50 +0000 |
| commit | dea2d2d58fe3a742e6f66fe50befbe0044e15ad1 (patch) | |
| tree | 87c30d892f45b8cc2de5fdb2c825d9ff05de7fae /src/runtime/CL/functions/CLPadLayer.cpp | |
| parent | 053e7510f24c2b02f9fae9c45fb6b874631a5376 (diff) | |
| download | ComputeLibrary-dea2d2d58fe3a742e6f66fe50befbe0044e15ad1.tar.gz | |
COMPMID-1772: Implement PadV2 for NEON
Change-Id: Ia4604524a034c46b004fd850183480c5fbfd8cb3
Reviewed-on: https://review.mlplatform.org/437
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLPadLayer.cpp')
| -rw-r--r-- | src/runtime/CL/functions/CLPadLayer.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
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{}; |