diff options
Diffstat (limited to 'ethosu/vela/weight_compressor.py')
-rw-r--r-- | ethosu/vela/weight_compressor.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py index 188b16ad..60ca846c 100644 --- a/ethosu/vela/weight_compressor.py +++ b/ethosu/vela/weight_compressor.py @@ -521,6 +521,9 @@ def update_pass_weight_and_scale_tensors(nng, arch): compress_weights( arch, nng, tens, op.type.npu_block_type, ps.block_config[-1], ofm_depth_step, op.get_dilation_h_w() ) + nng.total_compressed_weights += tens.weight_compressed_offsets[-1] + nng.total_original_weights += tens.elements() * tens.element_size() + # Update source tensor if needs_dma: src_tens = tens.get_dma_src_tensor() |