commit 79a89048a7c500384b66535c5ba77d31ee6cbdde
author wilisa01 <william.isaksson@arm.com> Wed Nov 02 17:18:43 2022 +0000
committer wilisa01 <william.isaksson@arm.com> Thu Nov 17 12:33:05 2022 +0000
tree b13302c2a6e8fb3d705c5098a8a47c455938c060
parent bc6ee58dbbd50fe83adfb1726c868898e5cfb491

MLBEDSW-6915: MLCE - Missing operators in Debug DB

- Adds missing operators and type conversion recording to DebugDB

Change-Id: If76b0b430bbe73ae1469024c3160ecf0eea26abe
Signed-off-by: wilisa01 <william.isaksson@arm.com>

ethosu/vela/graph_optimiser_util.py

diff --git a/ethosu/vela/graph_optimiser_util.py b/ethosu/vela/graph_optimiser_util.py
index fde01cf..4e233c4 100644
--- a/ethosu/vela/graph_optimiser_util.py
+++ b/ethosu/vela/graph_optimiser_util.py
@@ -257,7 +257,6 @@
         cons_op.read_shapes[1] = op.read_shapes[0]
         cons_op.set_input_tensor(op.ifm, cons_op.type.info.indices.ifms[1])
         cons_op.ifm_shapes[1] = op.ifm_shapes[0]
-
     op.ofm.consumer_list.remove(cons_op)
     op.ofm.ops = []
     op.ifm.consumer_list.remove(op)
@@ -270,7 +269,7 @@
 
 
 def record_optimised(op, arch):
-    if op.type != Op.Const:
+    if op.type not in (Op.Const, Op.Placeholder):
         DebugDatabase.add_optimised(op, op)
 
 
@@ -392,12 +391,12 @@
 
             weight_tensor.values = np.transpose(weight_tensor.values, (0, 1, 3, 2))
             weight_tensor.set_all_shapes(list(weight_tensor.values.shape))
+            DebugDatabase.add_optimised(op, op)
         else:
             raise UnsupportedFeatureError(
                 f"Unsupported 'DEPTHWISE_CONV_2D' with depth_multiplier = {op.attrs['depth_multiplier']},",
                 f" ifm channels = {ifm_shape.depth}, ofm channels = {ofm_shape.depth}",
             )
-        DebugDatabase.add_optimised(op, op)
     return op
 
 
@@ -426,4 +425,5 @@
     lut_tensor = lut.create_lut_tensor(op.name + "_values", lut_values, DataType.int8)
     op.set_activation_lut(lut_tensor)
     op.set_ifm_ofm_shapes()
+    DebugDatabase.add_optimised(op, op)
     return op