diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-20 10:47:47 +0200 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-20 13:27:15 +0200 |
commit | 46408a8049f6a51dda5bfa8a4c9959e037120265 (patch) | |
tree | 68595457843f3ff4193da0542afbe5de56da8d31 /ethosu/vela/tensor.py | |
parent | f436ada9caea87ec2dd686a92e41a15c1dcdeb1d (diff) | |
download | ethos-u-vela-46408a8049f6a51dda5bfa8a4c9959e037120265.tar.gz |
TOSA: Elementwise Rank > 4 and Batch > 1
Added support for elementwise operations:
-Support for up to Rank == 6
-Support for Batch > 1 for Rank == 4
-For binary elementwise ops this includes handling
of broadcasting in dimensions above H-dimension
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I73850bbfb288077a99bd2ceecbf989172016da24
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 37fd06ea..2e70d72e 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -632,7 +632,7 @@ class Tensor: self, coord: Optional[Shape] = None, shape4D: Optional[Shape4D] = None ) -> Tuple[Optional[Shape], Optional[Shape]]: if coord is None: - coord = [0] * len(self.storage_shape) + coord = [0] * min(len(self.storage_shape), 4) if shape4D and self.is_standard_fm: augmented_shape = self.get_4D_storage_shape_for_shape(shape4D).as_list() |