commit ef3ebddb6b4fd951046cbe7799721f6f0ed2fc87
author Patrik Gustavsson <patrik.gustavsson@arm.com> Fri Oct 01 11:10:25 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Fri Oct 01 11:03:38 2021 +0000
tree 9019c0cb4bb1a513ee1b4b00932e16031ab7da65
parent 3ff7a4aed60fefababdced93a5aba91409a4050c

TOSA: Add support for Identity operation

Added support for Identity operation.

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: If00b30528932f7531807ce3914d6c1875ab72fa4

ethosu/vela/tosa_graph_optimiser.py

diff --git a/ethosu/vela/tosa_graph_optimiser.py b/ethosu/vela/tosa_graph_optimiser.py
index d32955d..954ac68 100644
--- a/ethosu/vela/tosa_graph_optimiser.py
+++ b/ethosu/vela/tosa_graph_optimiser.py
@@ -182,7 +182,7 @@
 
 
 def fix_sg_input_output_tosa(op, arch, nng):
-    if not op.run_on_npu or op.type != Op.Reshape:
+    if not op.run_on_npu or op.type not in (Op.Reshape, Op.Identity):
         return op
 
     # For the Reshape operators we want to remove, tensors are removed.
@@ -306,8 +306,8 @@
     assert op.ofm_shapes[0][axis_4D] == offset
 
 
-def remove_reshapes(op, arch):
-    if op.run_on_npu and op.type == Op.Reshape:
+def remove_memory_ops(op, arch):
+    if op.run_on_npu and op.type in (Op.Reshape, Op.Identity):
         bypass_memory_only_ops(op)
 
 
@@ -820,7 +820,7 @@
 
     # Removal of reshapes
     for sg in nng.subgraphs:
-        rewrite_graph.visit_graph_post_order(sg.output_tensors, arch, [], [remove_reshapes])
+        rewrite_graph.visit_graph_post_order(sg.output_tensors, arch, [], [remove_memory_ops])
         sg.refresh_after_modification()
 
     # Decomposing of elementwise