aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/graph_optimiser.py
diff options
context:
space:
mode:
authorFredrik Svedberg <fredrik.svedberg@arm.com>2021-05-06 09:53:31 +0200
committerpatrik.gustavsson <patrik.gustavsson@arm.com>2021-05-07 11:09:28 +0000
commitc875aa6fdd8740f759305ff0fec9917977d019f0 (patch)
treed2ceca0ca8816a59fa029fdc39ec826d05533bb6 /ethosu/vela/graph_optimiser.py
parente6b94bb0a6d6021c5e575aef80fbcaaa332d7e33 (diff)
downloadethos-u-vela-c875aa6fdd8740f759305ff0fec9917977d019f0.tar.gz
[MLBEDSW-4530] Improve --verbose-graph output
Improved --verbose-graph output by adding labels to each print. Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com> Change-Id: I49039ff6af1c06f49208591f02effa4ff73f982a
Diffstat (limited to 'ethosu/vela/graph_optimiser.py')
-rw-r--r--ethosu/vela/graph_optimiser.py4
1 files changed, 2 insertions, 2 deletions
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