diff options
Diffstat (limited to 'ethosu/vela/graph_optimiser.py')
-rw-r--r-- | ethosu/vela/graph_optimiser.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index 50368b86..f5006c6f 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -109,6 +109,7 @@ def rewrite_concat_ops(op, arch): DebugDatabase.add_optimised(op, avgpool_op) avgpool_op.ifm_shapes.append(op.ifm_shapes[idx]) avgpool_op.ofm_shapes.append(op.ofm_shapes[0]) + avgpool_op.memory_function = Op.ConcatSliceWrite assert ofm.shape[axis] == offset # If axis corresponds to C-dimension, NHCWB16 can only be used in the output if all the concat_start's are a |