diff options
Diffstat (limited to 'src/runtime/NEON')
-rw-r--r-- | src/runtime/NEON/INEOperator.cpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/runtime/NEON/INEOperator.cpp b/src/runtime/NEON/INEOperator.cpp index ccee8ffc21..a5fc0a2726 100644 --- a/src/runtime/NEON/INEOperator.cpp +++ b/src/runtime/NEON/INEOperator.cpp @@ -44,7 +44,12 @@ void INEOperator::run(ITensorPack &tensors) 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) |