diff options
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 42ba853d..1a071e61 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -690,6 +690,11 @@ class Tensor: return True return False + def set_all_shapes(self, shape): + self.shape = shape + self.storage_shape = shape + self.bandwidth_shape = shape + def __str__(self): return "<nng.Tensor '%s' shape=%s dtype=%s>" % (self.name, self.shape, self.dtype) |