diff options
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): |