diff options
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 008cd05e..86306cad 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -501,10 +501,9 @@ class Tensor: return res - def clone_into_fast_storage(self, arch) -> "Tensor": - res = self.clone(suffix="_fast_storage") - res.mem_area = arch.fast_storage_mem_area - res.mem_type = MemType.Scratch_fast + def clone_into_shram(self, arch) -> "Tensor": + res = self.clone(suffix="_shram") + res.mem_area = MemArea.Shram res.src_tensor = self return res |