diff options
-rw-r--r-- | ethosu/vela/architecture_features.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/architecture_features.py b/ethosu/vela/architecture_features.py index 822bc115..021597e6 100644 --- a/ethosu/vela/architecture_features.py +++ b/ethosu/vela/architecture_features.py @@ -268,6 +268,7 @@ Note the difference between ArchitectureFeatures and CompilerOptions } self.tensor_storage_mem_type = { + TensorPurpose.Unknown: MemType.Unknown, TensorPurpose.Weights: MemType.Permanent_NPU, TensorPurpose.FeatureMap: MemType.Scratch, TensorPurpose.LUT: MemType.Scratch, |