diff options
| author | Manuel Bottini <manuel.bottini@arm.com> | 2020-06-15 16:50:35 +0100 |
|---|---|---|
| committer | Manuel Bottini <manuel.bottini@arm.com> | 2020-07-01 09:17:36 +0000 |
| commit | f2c714b244d0cdd8c38816bcc6b9e7eb3be7ee66 (patch) | |
| tree | ccc953def7bcf52ddefd34afb87cd25152a72320 /src/runtime/NEON/functions/NENormalizationLayer.cpp | |
| parent | f94c674a446d688aa4091e22402b3a7dc9cc5cc5 (diff) | |
| download | ComputeLibrary-f2c714b244d0cdd8c38816bcc6b9e7eb3be7ee66.tar.gz | |
COMPMID-3153: Remove padding from NENormalizationLayerKernel
Change-Id: Ib84308ea18bfa31ffbc3269a1f005d7d302139f7
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3350
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NENormalizationLayer.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NENormalizationLayer.cpp | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/src/runtime/NEON/functions/NENormalizationLayer.cpp b/src/runtime/NEON/functions/NENormalizationLayer.cpp index d52e92828e..f3a3ac6322 100644 --- a/src/runtime/NEON/functions/NENormalizationLayer.cpp +++ b/src/runtime/NEON/functions/NENormalizationLayer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2019 ARM Limited. + * Copyright (c) 2017-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -30,10 +30,10 @@ #include "arm_compute/core/Validate.h" #include "arm_compute/runtime/NEON/NEScheduler.h" -using namespace arm_compute; - +namespace arm_compute +{ NENormalizationLayer::NENormalizationLayer(std::shared_ptr<IMemoryManager> memory_manager) - : _memory_group(std::move(memory_manager)), _norm_kernel(), _multiply_kernel(), _border_handler(), _input_squared() + : _memory_group(std::move(memory_manager)), _norm_kernel(), _multiply_kernel(), _input_squared() { } @@ -50,7 +50,6 @@ void NENormalizationLayer::configure(const ITensor *input, ITensor *output, cons // Configure kernels _norm_kernel.configure(input, &_input_squared, output, norm_info); _multiply_kernel.configure(input, input, &_input_squared, 1.0f, ConvertPolicy::SATURATE, RoundingPolicy::TO_ZERO); - _border_handler.configure(&_input_squared, _norm_kernel.border_size(), BorderMode::CONSTANT, PixelValue(0.0f)); // Allocate the tensor once the configure methods have been called _input_squared.allocator()->allocate(); @@ -72,6 +71,6 @@ void NENormalizationLayer::run() MemoryGroupResourceScope scope_mg(_memory_group); NEScheduler::get().schedule(&_multiply_kernel, Window::DimY); - NEScheduler::get().schedule(&_border_handler, Window::DimY); NEScheduler::get().schedule(&_norm_kernel, Window::DimY); } +}
\ No newline at end of file |