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/vela.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/vela.py')
-rw-r--r-- | ethosu/vela/vela.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index c9551861..aa74ecf3 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -77,7 +77,12 @@ def process(input_name, enable_debug_db, arch, model_reader_options, compiler_op summary_csv_file = "{0}_summary_{1}.csv".format(output_basename, arch.system_config) stats_writer.write_summary_metrics_csv(nng, summary_csv_file, arch) - stats_writer.print_performance_metrics(nng, show_cpu_operations=compiler_options.show_cpu_operations, arch=arch) + stats_writer.print_performance_metrics( + nng, + show_cpu_operations=compiler_options.show_cpu_operations, + verbose_weights=compiler_options.verbose_weights, + arch=arch, + ) output_filename = output_basename + "_vela.tflite" if input_name.endswith(".tflite"): @@ -284,6 +289,7 @@ def main(args=None): "--verbose-register-command-stream", action="store_true", help="Verbose register command stream" ) parser.add_argument("--verbose-operators", action="store_true", help="Verbose operator list") + parser.add_argument("--verbose-weights", action="store_true", help="Verbose weights information") parser.add_argument( "--show-cpu-operations", action="store_true", help="Show the operations that fall back to the CPU" ) @@ -456,6 +462,7 @@ def main(args=None): verbose_high_level_command_stream=args.verbose_high_level_command_stream, verbose_register_command_stream=args.verbose_register_command_stream, verbose_operators=args.verbose_operators, + verbose_weights=args.verbose_weights, show_cpu_operations=args.show_cpu_operations, tensor_allocator=args.tensor_allocator, timing=args.timing, |