diff options
Diffstat (limited to 'ethosu/vela/mark_tensors.py')
-rw-r--r-- | ethosu/vela/mark_tensors.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py index c1572f4a..f3d5e855 100644 --- a/ethosu/vela/mark_tensors.py +++ b/ethosu/vela/mark_tensors.py @@ -41,6 +41,7 @@ def mark_purpose(tens, arch, purpose): tens.purpose = purpose elif tens.purpose not in (purpose, TensorPurpose.LUT): assert 0, "Cannot resolve tensor purpose {} and {} for tensor {}".format(tens.purpose, purpose, tens) + fmt = get_format(purpose, arch) tens.set_format(fmt, arch) tens.mem_area = arch.tensor_storage_mem_area[tens.purpose] |