MLBEDSW-3222: Bias tensors in fast storage
For IFM streamed cascades bias tensors are read several times.
Moves these tensors to fast storage and add DMA commands.
Change-Id: I630f6275986c1b5e3f126c925b11e22500fb1128
Signed-off-by: Andreas Nevalainen <andreas.nevalainen@arm.com>
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 49f93cd..45518b4 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -81,16 +81,17 @@
FeatureMap = 2
Scratch = 3
LUT = 4
- Size = 5
+ FSBias = 5
+ Size = 6
def display_name(self):
- return ("Unknown", "Weights", "FeatureMap", "Scratch", "LUT", "Size")[self.value]
+ return ("Unknown", "Weights", "FeatureMap", "Scratch", "LUT", "FastStorageBias", "Size")[self.value]
def identifier_name(self):
- return ("unknown", "weights", "feature_map", "scratch", "lut", "size")[self.value]
+ return ("unknown", "weights", "feature_map", "scratch", "lut", "fast_storage_bias", "size")[self.value]
def all():
- return (TensorPurpose.Weights, TensorPurpose.FeatureMap)
+ return (TensorPurpose.Weights, TensorPurpose.FeatureMap, TensorPurpose.FSBias)
class TensorSubPurpose(enum.Enum):