diff options
Diffstat (limited to 'ethosu/vela/tensor.py')
| -rw-r--r-- | ethosu/vela/tensor.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 312e8f35..c41a7ebf 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -626,7 +626,7 @@ class Tensor: for op in self.consumers(): if op.type == "DMA": return op.outputs[0].find_npu_op() - if "npu_block_type" in op.attrs: + if op.run_on_npu: return op return None |