diff options
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); } |