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 fab00e00..8e108dbf 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -456,8 +456,7 @@ def pack_into_passes(nng, arch, verbose_packing=False): avgpool_op.attrs["explicit_padding"] = [0, 0, 0, 0] avgpool_out = inp.clone("_avgpooled") avgpool_out.consumer_list.append(op) - avgpool_out.ops = [avgpool_op] - avgpool_op.outputs = [avgpool_out] + avgpool_op.set_output_tensor(avgpool_out) op.inputs[0] = avgpool_out ops_list.insert(0, avgpool_op) |