diff options
| author | Giorgio Arena <giorgio.arena@arm.com> | 2021-08-16 12:29:27 +0100 |
|---|---|---|
| committer | Giorgio Arena <giorgio.arena@arm.com> | 2021-08-19 10:46:51 +0000 |
| commit | 73df9310f4e94e43597f283307e3cde0653d8bae (patch) | |
| tree | b0a286dc00496cde6b98c1f96119ae5c830b1815 /arm_compute/runtime/IWeightsManager.h | |
| parent | df5bcb67e895610c0a7360753817fea13faabd4e (diff) | |
| download | ComputeLibrary-73df9310f4e94e43597f283307e3cde0653d8bae.tar.gz | |
Address comments on avoiding releasing weights if used by multiple functions
Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: I0b59c5326f5fcbc322fbeb864197ea999de6bd56
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6112
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/IWeightsManager.h')
| -rw-r--r-- | arm_compute/runtime/IWeightsManager.h | 4 |
1 files changed, 2 insertions, 2 deletions
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 |