diff options
Diffstat (limited to 'src/runtime/NEON/INEOperator.cpp')
| -rw-r--r-- | src/runtime/NEON/INEOperator.cpp | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/src/runtime/NEON/INEOperator.cpp b/src/runtime/NEON/INEOperator.cpp index ccee8ffc21..fcfd3251ff 100644 --- a/src/runtime/NEON/INEOperator.cpp +++ b/src/runtime/NEON/INEOperator.cpp @@ -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, _kernel->window(), 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) |