diff options
| author | Johan Alfvén <johan.alfven@arm.com> | 2022-10-21 11:21:38 +0200 |
|---|---|---|
| committer | Johan Alfvén <johan.alfven@arm.com> | 2022-10-25 15:31:08 +0200 |
| commit | 0f2e59f8bff0ca68794db1406e1264531da1d3a5 (patch) | |
| tree | ef3c4420dd5fe47c9cd5cc1cd6bb79082962b923 /ethosu/vela/tensor.py | |
| parent | 2a285fca30d13f6577ef3e8154aea24713d728a5 (diff) | |
| download | ethos-u-vela-0f2e59f8bff0ca68794db1406e1264531da1d3a5.tar.gz | |
MLBEDSW-7028: Fix compiler assert for elementwise op
- Refactored erroneously if statement that allowed illegal
swapping between ifm1 and ifm2 for elementwise operators.
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Change-Id: Iec571f710824432edac9104d960f199f33a1b241
Diffstat (limited to 'ethosu/vela/tensor.py')
| -rw-r--r-- | ethosu/vela/tensor.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 17d41b1a..ba385886 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -436,6 +436,17 @@ class Tensor: def is_standard_fm(self) -> bool: return self.sub_purpose == TensorSubPurpose.Standard and self.purpose == TensorPurpose.FeatureMap + @property + def is_const(self) -> bool: + return self.ops != [] and self.ops[0].type == Op.Const + + @property + def is_scalar(self) -> bool: + return self.shape == [] and self.elements() == 1 + + def is_broadcast(self, ofm) -> bool: + return self.shape != ofm.shape + def element_size(self) -> int: if self.element_size_bytes == 0: return self.dtype.size_in_bits() // 8 |