diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEGEMMConv2d.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEGEMMConv2d.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEGEMMConv2d.cpp b/src/runtime/NEON/functions/NEGEMMConv2d.cpp index 3ca5239ae3..7e2ce70444 100644 --- a/src/runtime/NEON/functions/NEGEMMConv2d.cpp +++ b/src/runtime/NEON/functions/NEGEMMConv2d.cpp @@ -104,12 +104,12 @@ void NEGEMMConv2d::prepare() // Release temporary tensors that are only used in prepare stage for(auto &ws : _impl->workspace) { - const int slot = ws.first; + const int slot = ws.slot; for(auto &m : _impl->aux_mem_req) { if(m.slot == slot && m.lifetime == MemoryLifetime::Prepare) { - auto tensor = ws.second.get(); + auto tensor = ws.tensor.get(); tensor->allocator()->free(); break; } |