diff options
Diffstat (limited to 'ethosu/vela/graph_optimiser.py')
| -rw-r--r-- | ethosu/vela/graph_optimiser.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index daae915d..26ee73c2 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -108,6 +108,10 @@ def rewrite_split(tens, arch): break offset_start[axis] += out.shape[axis] + # If start offset is not a multiple of 16 in the C-dimension, NHCWB16 need to be avoided in the input + if (offset_start[-1] % 16) != 0: + inp.avoid_NHCWB16 = True + offset_end[axis] = offset_start[axis] + tens.shape[axis] new_op.attrs["split_start"] = offset_start |