| 152 | } |
| 153 | |
| 154 | void IWeightsManager::release(const ITensor *weights) |
| 155 | { |
| 156 | if (weights == nullptr || !are_weights_managed(weights)) |
| 157 | { |
| 158 | return; |
| 159 | } |
| 160 | |
| 161 | _managed_counter[weights].counter--; |
| 162 | if (_managed_counter[weights].counter == 0 && _managed_counter[weights].is_unused) |
| 163 | { |
| 164 | weights->mark_as_unused(); |
| 165 | } |
| 166 | } |
| 167 | |
| 168 | void IWeightsManager::pre_mark_as_unused(const ITensor *weights) |
| 169 | { |