diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEGaussian5x5.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEGaussian5x5.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEGaussian5x5.cpp b/src/runtime/NEON/functions/NEGaussian5x5.cpp index f085975b1e..b010ca0104 100644 --- a/src/runtime/NEON/functions/NEGaussian5x5.cpp +++ b/src/runtime/NEON/functions/NEGaussian5x5.cpp @@ -57,9 +57,10 @@ void NEGaussian5x5::configure(ITensor *input, ITensor *output, BorderMode border void NEGaussian5x5::run() { + NEScheduler::get().schedule(&_border_handler, Window::DimZ); + _memory_group.acquire(); - NEScheduler::get().schedule(&_border_handler, Window::DimZ); NEScheduler::get().schedule(&_kernel_hor, Window::DimY); NEScheduler::get().schedule(&_kernel_vert, Window::DimY); |