vela: Fix tensor purpose for some CPU only ops
- Add support for marking the tensor purpose of CPU only ops such as LESS which mark their input based upon their output
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: Ia7898089f0b18ccd4f183e2ef961a67f4d169e4c
diff --git a/ethosu/vela/architecture_features.py b/ethosu/vela/architecture_features.py
index b59122e..f1cabdd 100644
--- a/ethosu/vela/architecture_features.py
+++ b/ethosu/vela/architecture_features.py
@@ -243,6 +243,7 @@
self.tensor_storage_mem_area = {
# permanent mem_area
+ TensorPurpose.Unknown: MemArea.Unknown,
TensorPurpose.Weights: self.permanent_storage_mem_area,
TensorPurpose.FeatureMap: self.feature_map_storage_mem_area,
}