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/operation.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/operation.py')
| -rw-r--r-- | ethosu/vela/operation.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 1558b943..b4267926 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -545,6 +545,7 @@ class Operation: res.rounding_mode = self.rounding_mode res.explicit_scaling = self.explicit_scaling res.low_precision_scaling = self.low_precision_scaling + res.rescale = self.rescale return res |