commit e2bfa7ec7495591139e19f41397054a44a0348bc
author Patrik Gustavsson <patrik.gustavsson@arm.com> Wed Sep 08 15:04:11 2021 +0200
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Thu Sep 09 14:39:57 2021 +0200
tree 6f4616c1fd2f28ccabddae1489ef5f4cad24c464
parent f366fb1fdffcc9d0eb2e6daf60dc89a2bd442ce6

TOSA: Add support for PAD

Added support for TOSA PAD operator
in line with legacy support
Limitations:
-Rank <= 4
-N = 1 if Rank = 4 for ifms/ofm
-only padding in W and H dimensions
-bool_t not supported

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I511608202b4c9bf6d86285b559c517fb41741fdf

ethosu/vela/operation.py

diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 681f498..e9d364e 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -592,6 +592,9 @@
     def get_ifm_ifm2_weights_ofm(self):
         return self.ifm, self.ifm2, self.weights, self.ofm
 
+    def get_ifm_ifm2_ofm(self):
+        return self.ifm, self.ifm2, self.ofm
+
     def get_ifm_weights_biases_ofm(self):
         return self.ifm, self.weights, self.bias, self.ofm