commit 2349d429d926e258e9a61d34c7fd97660ab9fb98
author Patrik Gustavsson <patrik.gustavsson@arm.com> Tue Dec 01 16:02:29 2020 +0100
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Fri Dec 18 16:33:32 2020 +0100
tree b5151d0f12428e47d64b1fb2ce4f2f8c19304a0d
parent 528a56df829b65f7a2c61953650b123c461095f7

MLBEDSW-3654 Add/use op ifm/ofm shapes

Add ifm/ofm shapes to op
Changed to rely on these shapes

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

ethosu/vela/live_range.py

diff --git a/ethosu/vela/live_range.py b/ethosu/vela/live_range.py
index 14e83a3..0cc89e2 100644
--- a/ethosu/vela/live_range.py
+++ b/ethosu/vela/live_range.py
@@ -181,12 +181,12 @@
                 inps.append(elem_op.ifm2)
 
             if len(inps) > 0:
-                for inp in inps:
+                for i, inp in enumerate(inps):
                     # check input format, dtype, broadcasting or if there are more input consumers
                     if (
                         inp.format == elem_op.ofm.format
                         and inp.dtype == elem_op.ofm.dtype
-                        and inp.shape == elem_op.ofm.shape
+                        and elem_op.ifm_shapes[i] == elem_op.ofm_shapes[0]
                         and (len(inp.consumer_list) == 1 and len(inp.ops) == 1)
                     ):
                         lr_graph.fuse_ranges(inp, elem_op.ofm)