diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-01-27 16:53:41 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-01-28 09:08:43 +0000 |
commit | e22ba8cb3090886b2d80a2df0e599dbf4cd7f483 (patch) | |
tree | c980972830ec15613ab291c7731968ae1a010609 /ethosu/vela/tensor.py | |
parent | 60232140a2927865d1f6f9bc48871df3b2bb135b (diff) | |
download | ethos-u-vela-e22ba8cb3090886b2d80a2df0e599dbf4cd7f483.tar.gz |
[MLBEDSW-3891] Fix reading back in an ethos-u custom op
Fixed assertion when reading back in an ethos-u custom op.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I275ec9187ffead1e96f2522ecbd658328fa4ef69
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 093e8771..fb877ca8 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -111,15 +111,20 @@ class TensorPurpose(enum.IntFlag): Weights = 1 FeatureMap = 2 Scratch = 3 - LUT = 4 - FSBias = 5 - Size = 6 + ScratchFast = 4 + LUT = 5 + FSBias = 6 + Size = 7 def display_name(self) -> str: - return ("Unknown", "Weights", "FeatureMap", "Scratch", "LUT", "FastStorageBias", "Size")[self.value] + return ("Unknown", "Weights", "FeatureMap", "Scratch", "ScratchFast", "LUT", "FastStorageBias", "Size")[ + self.value + ] def identifier_name(self) -> str: - return ("unknown", "weights", "feature_map", "scratch", "lut", "fast_storage_bias", "size")[self.value] + return ("unknown", "weights", "feature_map", "scratch", "scratch_fast", "lut", "fast_storage_bias", "size")[ + self.value + ] @staticmethod def all(): |