diff options
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 998d94ff..31839c72 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -852,6 +852,11 @@ class Operation: return self.forced_input_quantization return self.ifm.quantization + def add_output_tensor(self, tens): + self.outputs.append(tens) + if self not in tens.ops: + tens.ops.append(self) + def set_output_tensor(self, tens): tens.ops = [self] self.outputs = [tens] |