aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tensor_allocation.py
diff options
context:
space:
mode:
authorFredrik Svedberg <fredrik.svedberg@arm.com>2021-04-23 14:36:42 +0200
committerpatrik.gustavsson <patrik.gustavsson@arm.com>2021-05-06 12:52:21 +0000
commitf5c07c45b48717fb6391adb35cb73ca7cd5734c3 (patch)
tree199fc4646a403af01009c8e10aafec0bad1e910e /ethosu/vela/tensor_allocation.py
parenta0b8d5f1bc32092cb85df07fb68e5582d01def32 (diff)
downloadethos-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/tensor_allocation.py')
-rw-r--r--ethosu/vela/tensor_allocation.py4
1 files changed, 0 insertions, 4 deletions
diff --git a/ethosu/vela/tensor_allocation.py b/ethosu/vela/tensor_allocation.py
index 0ad30e5f..7ffc6f3d 100644
--- a/ethosu/vela/tensor_allocation.py
+++ b/ethosu/vela/tensor_allocation.py
@@ -217,9 +217,5 @@ def allocate_tensors(
if sg == nng.get_root_subgraph():
nng.memory_used = sg.memory_used
- try:
- nng.weights_compression_ratio = nng.total_compressed_weights / nng.total_original_weights
- except ZeroDivisionError:
- nng.weights_compression_ratio = 0.0
return True