From c875aa6fdd8740f759305ff0fec9917977d019f0 Mon Sep 17 00:00:00 2001 From: Fredrik Svedberg Date: Thu, 6 May 2021 09:53:31 +0200 Subject: [MLBEDSW-4530] Improve --verbose-graph output Improved --verbose-graph output by adding labels to each print. Signed-off-by: Fredrik Svedberg Change-Id: I49039ff6af1c06f49208591f02effa4ff73f982a --- ethosu/vela/graph_optimiser.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'ethosu/vela/graph_optimiser.py') diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index 99d4bf05..f4472f9e 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -1680,7 +1680,7 @@ def _record_optimised(op, arch): def optimise_graph_a(nng, arch, verbose_graph=False): if verbose_graph: - nng.print_graph() + nng.print_graph("Before Graph Optimization") pre_process_list = [ supported_operator_check, @@ -1783,5 +1783,5 @@ def optimise_graph_a(nng, arch, verbose_graph=False): rewrite_graph.visit_graph_post_order(sg.output_tensors, arch, [], [check_reshapes, _record_optimised]) if verbose_graph: - nng.print_graph() + nng.print_graph("After Graph Optimization") return nng -- cgit v1.2.1