From 73df9310f4e94e43597f283307e3cde0653d8bae Mon Sep 17 00:00:00 2001 From: Giorgio Arena Date: Mon, 16 Aug 2021 12:29:27 +0100 Subject: Address comments on avoiding releasing weights if used by multiple functions Signed-off-by: Giorgio Arena Change-Id: I0b59c5326f5fcbc322fbeb864197ea999de6bd56 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6112 Tested-by: Arm Jenkins Reviewed-by: Gian Marco Iodice Reviewed-by: SiCong Li Comments-Addressed: Arm Jenkins --- arm_compute/runtime/IWeightsManager.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'arm_compute/runtime') diff --git a/arm_compute/runtime/IWeightsManager.h b/arm_compute/runtime/IWeightsManager.h index db39a71314..3b97d696bb 100644 --- a/arm_compute/runtime/IWeightsManager.h +++ b/arm_compute/runtime/IWeightsManager.h @@ -81,11 +81,11 @@ public: * @param[in] weights Weights to release */ void release(const ITensor *weights); - /** Marks weights as unused + /** Pre-mark the weights as unused. The weights tensor will get marked as unused only when the counter goes to 0 * * @param weights Weights to mark unused */ - void mark_as_unused(const ITensor *weights); + void pre_mark_as_unused(const ITensor *weights); private: struct CounterElement -- cgit v1.2.1