diff options
Diffstat (limited to 'ethosu/vela/pass_packing.py')
| -rw-r--r-- | ethosu/vela/pass_packing.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 518b2436..b28f4eb4 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -348,8 +348,7 @@ def pack_into_passes(nng, arch, verbose_packing=False): ps.ifm_shapes.append(op.ifm_shapes[0]) elif ps.ifm_tensor == op.ifm2: ps.ifm_shapes.append(op.ifm_shapes[1]) - for op in input_ops_list + [primary_op]: - if op.run_on_npu: + if ps.ifm2_tensor == op.ifm: ps.ifm_shapes.append(op.ifm_shapes[0]) elif ps.ifm2_tensor == op.ifm2: |