commit f5c07c45b48717fb6391adb35cb73ca7cd5734c3
author Fredrik Svedberg <fredrik.svedberg@arm.com> Fri Apr 23 14:36:42 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Thu May 06 12:52:21 2021 +0000
tree 199fc4646a403af01009c8e10aafec0bad1e910e
parent a0b8d5f1bc32092cb85df07fb68e5582d01def32

[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

ethosu/vela/compiler_driver.py

diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py
index a3c0100..26d350e 100644
--- a/ethosu/vela/compiler_driver.py
+++ b/ethosu/vela/compiler_driver.py
@@ -61,6 +61,7 @@
         verbose_high_level_command_stream=False,
         verbose_register_command_stream=False,
         verbose_operators=False,
+        verbose_weights=False,
         show_cpu_operations=False,
         tensor_allocator=TensorAllocator.Greedy,
         timing=False,
@@ -77,6 +78,7 @@
         self.verbose_high_level_command_stream = verbose_high_level_command_stream
         self.verbose_register_command_stream = verbose_register_command_stream
         self.verbose_operators = verbose_operators
+        self.verbose_weights = verbose_weights
         self.show_cpu_operations = show_cpu_operations
         self.tensor_allocator = tensor_allocator
         self.timing = timing