commit f5c07c45b48717fb6391adb35cb73ca7cd5734c3
author Fredrik Svedberg <fredrik.svedberg@arm.com> Fri Apr 23 14:36:42 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Thu May 06 12:52:21 2021 +0000
tree 199fc4646a403af01009c8e10aafec0bad1e910e
parent a0b8d5f1bc32092cb85df07fb68e5582d01def32

[MLBEDSW-4254] Improve weight information in summary

Improved weight information showed in summary if --verbose-weights
option is used.

Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: Iac142f2a813bf1c05aa9da3f8a384466e2914d06

ethosu/vela/nn_graph.py

diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index 2d4b0c8..677a385 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -517,9 +517,9 @@
         self.subgraphs = []
         self.metadata = []
         self.memory_used = {}
-        self.weights_compression_ratio = 0
         self.total_original_weights = 0
-        self.total_compressed_weights = 0
+        self.total_npu_weights = 0
+        self.total_npu_encoded_weights = 0
         self.weight_cache = None  # See CompressedWeightCache
 
     def get_root_subgraph(self):