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
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 17d41b1..ba38588 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -436,6 +436,17 @@
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