diff options
Diffstat (limited to 'ethosu/vela/npu_serialisation.py')
-rw-r--r-- | ethosu/vela/npu_serialisation.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py index f462168a..06ea61df 100644 --- a/ethosu/vela/npu_serialisation.py +++ b/ethosu/vela/npu_serialisation.py @@ -82,7 +82,7 @@ def serialise_npu_subgraph_into_tensors(sg, arch, scratch_tens, scratch_fast_ten sg.scratch_fast_tensor = make_memory_tensor( sg.name + "_scratch_fast", scratch_fast_area, MemType.Scratch_fast, 0, False, arch ) - sg.scratch_fast_tensor.purpose = TensorPurpose.Scratch + sg.scratch_fast_tensor.purpose = TensorPurpose.ScratchFast else: sg.scratch_tensor = scratch_tens sg.scratch_tensor.shape[0] += scratch_size |