diff options
Diffstat (limited to 'src/runtime/NEON/INEOperator.cpp')
| -rw-r--r-- | src/runtime/NEON/INEOperator.cpp | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/src/runtime/NEON/INEOperator.cpp b/src/runtime/NEON/INEOperator.cpp index a13b29b572..fcfd3251ff 100644 --- a/src/runtime/NEON/INEOperator.cpp +++ b/src/runtime/NEON/INEOperator.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020 Arm Limited. + * Copyright (c) 2020-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -22,8 +22,10 @@ * SOFTWARE. */ #include "arm_compute/runtime/NEON/INEOperator.h" + #include "arm_compute/core/Window.h" #include "arm_compute/runtime/NEON/NEScheduler.h" + #include "src/core/NEON/INEKernel.h" namespace arm_compute @@ -32,19 +34,23 @@ namespace experimental { INEOperator::~INEOperator() = default; -INEOperator::INEOperator(IRuntimeContext *ctx) - : _kernel(), _ctx(ctx), _workspace() +INEOperator::INEOperator(IRuntimeContext *ctx) : _kernel(), _ctx(ctx), _workspace() { } void INEOperator::run(ITensorPack &tensors) { - if(tensors.empty()) + if (tensors.empty()) { ARM_COMPUTE_ERROR("No inputs provided"); } - NEScheduler::get().schedule_op(_kernel.get(), Window::DimY, tensors); + run(tensors, _kernel->window()); +} + +void INEOperator::run(ITensorPack &tensors, const Window &window) +{ + NEScheduler::get().schedule_op(_kernel.get(), Window::DimY, window, tensors); } void INEOperator::prepare(ITensorPack &constants) @@ -54,7 +60,7 @@ void INEOperator::prepare(ITensorPack &constants) MemoryRequirements INEOperator::workspace() const { - return {}; + return _workspace; } } // namespace experimental } // namespace arm_compute |