diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEFillBorder.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEFillBorder.cpp | 22 |
1 files changed, 17 insertions, 5 deletions
diff --git a/src/runtime/NEON/functions/NEFillBorder.cpp b/src/runtime/NEON/functions/NEFillBorder.cpp index 6b7a0faa85..a3ab9c3db4 100644 --- a/src/runtime/NEON/functions/NEFillBorder.cpp +++ b/src/runtime/NEON/functions/NEFillBorder.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2016-2019 ARM Limited. + * Copyright (c) 2016-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -26,15 +26,27 @@ #include "arm_compute/core/Window.h" #include "arm_compute/runtime/NEON/NEScheduler.h" +#include "src/common/utils/Log.h" +#include "src/core/NEON/kernels/NEFillBorderKernel.h" + namespace arm_compute { -void NEFillBorder::configure(ITensor *input, unsigned int border_width, BorderMode border_mode, const PixelValue &constant_border_value) +NEFillBorder::NEFillBorder() : _border_handler(nullptr) +{ +} + +void NEFillBorder::configure(ITensor *input, + unsigned int border_width, + BorderMode border_mode, + const PixelValue &constant_border_value) { - _border_handler.configure(input, BorderSize(border_width), border_mode, constant_border_value); + ARM_COMPUTE_LOG_PARAMS(input, border_width, border_mode, constant_border_value); + _border_handler = std::make_unique<NEFillBorderKernel>(); + _border_handler->configure(input, BorderSize(border_width), border_mode, constant_border_value); } void NEFillBorder::run() { - NEScheduler::get().schedule(&_border_handler, Window::DimZ); + NEScheduler::get().schedule(_border_handler.get(), Window::DimZ); } -} // namespace arm_compute
\ No newline at end of file +} // namespace arm_compute |