diff options
Diffstat (limited to 'src/runtime/NEON')
-rw-r--r-- | src/runtime/NEON/INESimpleFunctionNoBorder.cpp | 10 | ||||
-rw-r--r-- | src/runtime/NEON/functions/NEActivationLayer.cpp | 12 |
2 files changed, 15 insertions, 7 deletions
diff --git a/src/runtime/NEON/INESimpleFunctionNoBorder.cpp b/src/runtime/NEON/INESimpleFunctionNoBorder.cpp index 12872048c7..2cabee4c46 100644 --- a/src/runtime/NEON/INESimpleFunctionNoBorder.cpp +++ b/src/runtime/NEON/INESimpleFunctionNoBorder.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018 ARM Limited. + * Copyright (c) 2018-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -24,16 +24,18 @@ #include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h" #include "arm_compute/runtime/NEON/NEScheduler.h" +#include "arm_compute/runtime/Utils.h" namespace arm_compute { -INESimpleFunctionNoBorder::INESimpleFunctionNoBorder() // NOLINT - : _kernel() +INESimpleFunctionNoBorder::INESimpleFunctionNoBorder(IRuntimeContext *ctx) + : _kernel(), + _ctx(ctx) { } void INESimpleFunctionNoBorder::run() { - NEScheduler::get().schedule(_kernel.get(), Window::DimY); + schedule_kernel_on_ctx(_ctx, _kernel.get(), Window::DimY); } } // namespace arm_compute diff --git a/src/runtime/NEON/functions/NEActivationLayer.cpp b/src/runtime/NEON/functions/NEActivationLayer.cpp index 6af71a3580..1b8651487d 100644 --- a/src/runtime/NEON/functions/NEActivationLayer.cpp +++ b/src/runtime/NEON/functions/NEActivationLayer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -24,10 +24,15 @@ #include "arm_compute/runtime/NEON/functions/NEActivationLayer.h" #include "arm_compute/core/NEON/kernels/NEActivationLayerKernel.h" +#include "arm_compute/runtime/IRuntimeContext.h" #include "support/ToolchainSupport.h" -using namespace arm_compute; - +namespace arm_compute +{ +NEActivationLayer::NEActivationLayer(IRuntimeContext *ctx) // NOLINT + : INESimpleFunctionNoBorder(ctx) +{ +} void NEActivationLayer::configure(ITensor *input, ITensor *output, ActivationLayerInfo activation_info) { auto k = arm_compute::support::cpp14::make_unique<NEActivationLayerKernel>(); @@ -39,3 +44,4 @@ Status NEActivationLayer::validate(const ITensorInfo *input, const ITensorInfo * { return NEActivationLayerKernel::validate(input, output, act_info); } +} // namespace arm_compute |