commit db5124c2b5e10b34c61b3e016bb597ba1c1574df
author Diqing Zhong <diqing.zhong@arm.com> Mon Jan 11 12:52:48 2021 +0100
committer Diqing Zhong <diqing.zhong@arm.com> Tue Jan 19 11:29:05 2021 +0100
tree 82bbd348c6a0ff9d3d4e6a44067c032922385d0b
parent 9a0cff1cd1334f4d3e7dfb542ad0be4f0e71a9de

MLBEDSW-3144: Add weights compression ratio

  - Also removed the original bit_per_element

Change-Id: I51bfbd28e14f316aae2d542bb610a3ed57b8b53b
Signed-off-by: Diqing Zhong <diqing.zhong@arm.com>

ethosu/vela/nn_graph.py

diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index d2c848a..71d4e61 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -512,9 +512,9 @@
         self.subgraphs = []
         self.metadata = []
         self.memory_used = {}
-        self.bits_per_element = {}
-        self.total_size = {}
-        self.total_elements = {}
+        self.weights_compression_ratio = 0
+        self.total_original_weights = 0
+        self.total_compressed_weights = 0
         self.weight_cache = None  # See CompressedWeightCache
 
     def get_root_subgraph(self):