diff options
Diffstat (limited to 'ethosu/vela/mark_tensors.py')
| -rw-r--r-- | ethosu/vela/mark_tensors.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py index 208b5b8c..a971ef23 100644 --- a/ethosu/vela/mark_tensors.py +++ b/ethosu/vela/mark_tensors.py @@ -367,6 +367,8 @@ def mark_tensor_format(nng, arch, verbose_tensor_format=False): visit_tens(tens, ps) for tens, fmt in formats_for_tensor.items(): + if len(tens.shape) > 4: + continue tens.set_format(fmt, arch) if fmt == TensorFormat.WeightsCompressed and tens.values is not None: src_tens = tens.get_dma_src_tensor() |