diff options
Diffstat (limited to 'ethosu/vela/mark_tensors.py')
| -rw-r--r-- | ethosu/vela/mark_tensors.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py index 40ce467b..03ab83fe 100644 --- a/ethosu/vela/mark_tensors.py +++ b/ethosu/vela/mark_tensors.py @@ -284,7 +284,7 @@ def mark_tensor_purpose(nng, arch, verbose_tensor_purpose=False): ) for idx, tens in enumerate(op.inputs): - purpose = input_purpose(op, idx) + purpose = input_purpose(op, idx) if tens.purpose == TensorPurpose.Unknown else tens.purpose mark_tensor_helper(tens, purpose) if op.type == "Reshape": |