diff options
| author | Manuel Bottini <manuel.bottini@arm.com> | 2020-06-01 14:40:54 +0100 |
|---|---|---|
| committer | Manuel Bottini <manuel.bottini@arm.com> | 2020-06-02 15:51:16 +0000 |
| commit | 16bd6dd97fb704e31cd96e404e8c4148fe24d834 (patch) | |
| tree | e326ea4817835de6ab5d9caa5a3a67c5c7bd2ad7 /src/core/NEON/kernels/NEROIPoolingLayerKernel.cpp | |
| parent | 7d6b5772c8854a470a08b3415eda1ceca531601d (diff) | |
| download | ComputeLibrary-16bd6dd97fb704e31cd96e404e8c4148fe24d834.tar.gz | |
COMPMID-3167: Remove padding from NEROIPoolingLayerKernel and NEROIAlignLayerKernel
Change-Id: I19106e4bb0f6886a23dccc5c71e864a8364f40f9
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3288
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NEROIPoolingLayerKernel.cpp')
| -rw-r--r-- | src/core/NEON/kernels/NEROIPoolingLayerKernel.cpp | 13 |
1 files changed, 4 insertions, 9 deletions
diff --git a/src/core/NEON/kernels/NEROIPoolingLayerKernel.cpp b/src/core/NEON/kernels/NEROIPoolingLayerKernel.cpp index 708420c67e..00c3f98334 100644 --- a/src/core/NEON/kernels/NEROIPoolingLayerKernel.cpp +++ b/src/core/NEON/kernels/NEROIPoolingLayerKernel.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2019 ARM Limited. + * Copyright (c) 2017-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -82,15 +82,10 @@ void NEROIPoolingLayerKernel::configure(const ITensor *input, const ITensor *roi window.set(Window::DimX, Window::Dimension(0, rois->info()->dimension(1))); window.set(Window::DimY, Window::Dimension(0, 1)); - AccessWindowStatic input_access(input->info(), - input->info()->valid_region().start(0), - input->info()->valid_region().start(1), - input->info()->valid_region().end(0), - input->info()->valid_region().end(1)); - AccessWindowStatic output_access(output->info(), 0, 0, pool_info.pooled_width(), pool_info.pooled_height()); + Coordinates coord; + coord.set_num_dimensions(output->info()->num_dimensions()); + output->info()->set_valid_region(ValidRegion(coord, output->info()->tensor_shape())); - ARM_COMPUTE_UNUSED(update_window_and_padding(window, input_access, output_access)); - output_access.set_valid_region(window, ValidRegion(Coordinates(), output->info()->tensor_shape())); INEKernel::configure(window); } |