aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/INEOperator.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/NEON/INEOperator.cpp')
-rw-r--r--src/runtime/NEON/INEOperator.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/NEON/INEOperator.cpp b/src/runtime/NEON/INEOperator.cpp
index ff643d1c43..ccee8ffc21 100644
--- a/src/runtime/NEON/INEOperator.cpp
+++ b/src/runtime/NEON/INEOperator.cpp
@@ -44,7 +44,7 @@ void INEOperator::run(ITensorPack &tensors)
ARM_COMPUTE_ERROR("No inputs provided");
}
- NEScheduler::get().schedule_op(_kernel.get(), Window::DimY, tensors);
+ NEScheduler::get().schedule_op(_kernel.get(), Window::DimY, _kernel->window(), tensors);
}
void INEOperator::prepare(ITensorPack &constants)