diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-04-08 12:02:58 +0100 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-04-19 13:45:08 +0000 |
| commit | 60c3b0e6821a80d78ffca5be30e05d062d071cd2 (patch) | |
| tree | 3e263a45aa9617cfd7704b2b33ea4337f1582321 /tests/NEON/Helper.h | |
| parent | 4f1650f0c9919f0bac5024b8e31c0f754d25aec3 (diff) | |
| download | ComputeLibrary-60c3b0e6821a80d78ffca5be30e05d062d071cd2.tar.gz | |
Port DepthwiseConvolution to new API
Resolves: COMPMID-4185
Change-Id: Ib5f22356356a022d567bb18d44ea272b62d10ebf
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5424
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/NEON/Helper.h')
| -rw-r--r-- | tests/NEON/Helper.h | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/tests/NEON/Helper.h b/tests/NEON/Helper.h index 714152ebcd..c9e53d11b0 100644 --- a/tests/NEON/Helper.h +++ b/tests/NEON/Helper.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2020 Arm Limited. + * Copyright (c) 2017-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -27,7 +27,9 @@ #include "arm_compute/runtime/Array.h" #include "arm_compute/runtime/NEON/INESimpleFunction.h" #include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h" +#include "arm_compute/runtime/NEON/NEScheduler.h" #include "src/core/NEON/kernels/NEFillBorderKernel.h" +#include "src/runtime/cpu/ICpuOperator.h" #include "tests/Globals.h" #include <algorithm> @@ -104,7 +106,7 @@ public: /** As above but this also setups a Zero border on the input tensor of the kernel's bordersize */ template <typename K> -class NESynthetizeFunctionWithZeroConstantKernelBorder : public INESimpleFunction +class NESynthetizeFunctionWithZeroConstantKernelBorder : public cpu::ICpuOperator { public: /** Configure the kernel. @@ -123,6 +125,15 @@ public: b->configure(first, BorderSize(_kernel->border_size()), BorderMode::CONSTANT, PixelValue()); _border_handler = std::move(b); } + + void run(ITensorPack &tensors) + { + NEScheduler::get().schedule(_border_handler.get(), Window::DimZ); + NEScheduler::get().schedule_op(_kernel.get(), Window::DimY, _kernel->window(), tensors); + } + +private: + std::unique_ptr<INEKernel> _border_handler{ nullptr }; }; } // namespace test |