commit 46408a8049f6a51dda5bfa8a4c9959e037120265
author Patrik Gustavsson <patrik.gustavsson@arm.com> Mon Sep 20 10:47:47 2021 +0200
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Mon Sep 20 13:27:15 2021 +0200
tree 68595457843f3ff4193da0542afbe5de56da8d31
parent f436ada9caea87ec2dd686a92e41a15c1dcdeb1d

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

ethosu/vela/operation.py

diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 1558b94..b426792 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -545,6 +545,7 @@
         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