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/shape4d.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/shape4d.py')
-rw-r--r-- | ethosu/vela/shape4d.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/shape4d.py b/ethosu/vela/shape4d.py index fd674031..08b2a6a0 100644 --- a/ethosu/vela/shape4d.py +++ b/ethosu/vela/shape4d.py @@ -136,6 +136,9 @@ class Shape4D(namedtuple("Shape4D", ["batch", "height", "width", "depth"])): def elements(self): return self.batch * self.width * self.height * self.depth + def dot_prod(self, rhs): + return self.batch * rhs.batch + self.width * rhs.width + self.height * rhs.height + self.depth * rhs.depth + def elements_wh(self): return self.width * self.height |