diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp b/src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp index 2600e2b946..427cd2eb77 100644 --- a/src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp +++ b/src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp @@ -78,11 +78,11 @@ void NEPoolingAssemblyDispatch::run(ITensorPack &tensors) if(_is_global_pooling_layer) { - NEScheduler::get().schedule_op(_kernel.get(), Window::DimX, tensors); + NEScheduler::get().schedule_op(_kernel.get(), Window::DimX, _kernel->window(), tensors); } else { - NEScheduler::get().schedule_op(_kernel.get(), Window::DimY, tensors); + NEScheduler::get().schedule_op(_kernel.get(), Window::DimY, _kernel->window(), tensors); } } } // namespace experimental |