diff options
author | Diqing Zhong <diqing.zhong@arm.com> | 2021-01-11 12:52:48 +0100 |
---|---|---|
committer | Diqing Zhong <diqing.zhong@arm.com> | 2021-01-19 11:29:05 +0100 |
commit | db5124c2b5e10b34c61b3e016bb597ba1c1574df (patch) | |
tree | 82bbd348c6a0ff9d3d4e6a44067c032922385d0b /ethosu/vela/nn_graph.py | |
parent | 9a0cff1cd1334f4d3e7dfb542ad0be4f0e71a9de (diff) | |
download | ethos-u-vela-db5124c2b5e10b34c61b3e016bb597ba1c1574df.tar.gz |
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>
Diffstat (limited to 'ethosu/vela/nn_graph.py')
-rw-r--r-- | ethosu/vela/nn_graph.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index d2c848ad..71d4e614 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -512,9 +512,9 @@ class Graph: 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): |