diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-04-23 14:36:42 +0200 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-05-06 12:52:21 +0000 |
commit | f5c07c45b48717fb6391adb35cb73ca7cd5734c3 (patch) | |
tree | 199fc4646a403af01009c8e10aafec0bad1e910e /ethosu/vela/nn_graph.py | |
parent | a0b8d5f1bc32092cb85df07fb68e5582d01def32 (diff) | |
download | ethos-u-vela-f5c07c45b48717fb6391adb35cb73ca7cd5734c3.tar.gz |
[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
Diffstat (limited to 'ethosu/vela/nn_graph.py')
-rw-r--r-- | ethosu/vela/nn_graph.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index 2d4b0c87..677a385a 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -517,9 +517,9 @@ class Graph: 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): |