aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp')
-rw-r--r--src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp b/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp
index 745179c050..b91048a426 100644
--- a/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp
+++ b/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp
@@ -99,12 +99,12 @@ void NEWinogradConvolutionLayer::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;
}