MLBEDSW-2528: MLCE-219: Custom operator pass through
- Fixed custom operator pass through
- Added error printing functions for operators and tensor
- Minor cleanup of custom exception handling
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: Idf295df1e4c544381dc480244d880c32fb285e38
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 51311ef..448d838 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -31,7 +31,7 @@
"""Class representing a Neural Network operation. Has a name, a type,
input and output tensors, as well as an attribute dictionary."""
- __slots__ = "type", "name", "attrs", "inputs", "outputs", "flops", "scheduled_pass", "run_on_npu"
+ __slots__ = "type", "name", "op_index", "attrs", "inputs", "outputs", "flops", "scheduled_pass", "run_on_npu"
def __init__(self, op_type, name):
self.type = op_type
@@ -42,6 +42,7 @@
self.flops = 0
self.run_on_npu = True
self.scheduled_pass = None
+ self.op_index = None # input network operator index
def clone(self, suffix="_clone"):
res = Operation(self.type, self.name + suffix)
@@ -51,6 +52,7 @@
res.outputs = list(self.outputs)
res.flops = self.flops
res.scheduled_pass = self.scheduled_pass
+ res.op_index = None # not relevant as not part of input network
return res