Port NEFullyConnectedLayer to memory injecting interface
Resolves: COMPMID-4501
Change-Id: Ib61b3d06974009e501b3fb86467735427e13a94a
Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5931
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
diff --git a/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp b/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp
index c405786..f63fcb0 100644
--- a/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp
+++ b/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp
@@ -105,19 +105,9 @@
{
_impl->weights->mark_as_unused();
}
- for(auto &ws : _impl->workspace_tensors)
- {
- const int slot = ws.slot;
- for(auto &m : _impl->aux_mem_req)
- {
- if(m.slot == slot && m.lifetime == MemoryLifetime::Prepare)
- {
- auto tensor = ws.tensor.get();
- tensor->allocator()->free();
- break;
- }
- }
- }
+
+ // Release temporary tensors that are only used in prepare stage
+ release_temporaries<Tensor>(_impl->aux_mem_req, _impl->workspace_tensors);
_impl->is_prepared = true;
}
}