From 46408a8049f6a51dda5bfa8a4c9959e037120265 Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Mon, 20 Sep 2021 10:47:47 +0200 Subject: 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 Change-Id: I73850bbfb288077a99bd2ceecbf989172016da24 --- ethosu/vela/operation.py | 1 + 1 file changed, 1 insertion(+) (limited to 'ethosu/vela/operation.py') 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 -- cgit v1.2.1