diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEReduceMean.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NEReduceMean.cpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEReduceMean.cpp b/src/runtime/NEON/functions/NEReduceMean.cpp index 014895f02b..98d3ab943d 100644 --- a/src/runtime/NEON/functions/NEReduceMean.cpp +++ b/src/runtime/NEON/functions/NEReduceMean.cpp @@ -137,7 +137,7 @@ void NEReduceMean::configure(ITensor *input, const Coordinates &reduction_axis, void NEReduceMean::run() { - _memory_group.acquire(); + MemoryGroupResourceScope scope_mg(_memory_group); for(unsigned int i = 0; i < _reduction_ops; ++i) { @@ -148,5 +148,4 @@ void NEReduceMean::run() { _reshape.run(); } - _memory_group.release(); } |