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/compiler_driver.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/compiler_driver.py')
| -rw-r--r-- | ethosu/vela/compiler_driver.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index a3c01001..26d350ea 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -61,6 +61,7 @@ Note the difference between ArchitectureFeatures and CompilerOptions 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 @@ Note the difference between ArchitectureFeatures and CompilerOptions 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 |