diff options
Diffstat (limited to 'ethosu/vela/operation.py')
| -rw-r--r-- | ethosu/vela/operation.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 7134fd82..adbbff51 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -311,3 +311,12 @@ input and output tensors, as well as an attribute dictionary.""" self.attrs["fused_activation_function"] = "LUT" self.activation_lut = lut_tensor self.inputs.append(lut_tensor) + + def add_input_tensor(self, tens): + self.inputs.append(tens) + if self not in tens.consumer_list: + tens.consumer_list.append(self) + + def set_output_tensor(self, tens): + tens.ops = [self] + self.outputs = [tens] |