commit b4936adf32435a96ca1a89818582f8d2efa366d1
author Patrik Gustavsson <patrik.gustavsson@arm.com> Tue Oct 05 13:53:34 2021 +0200
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Wed Oct 06 09:43:56 2021 +0200
tree b7b15246f1d705d5ba1b1c76eb841b412e11555a
parent 352607ca99a4a71de11a241227982b9c95dc69b2

TOSA: Added decomposition of PAD

Added support for:
-Rank > 4 and batch > 1
-Tensor dimensions exceeding NPU limit
-Padding in any dimension

(Implementation for functional compliance,
 not considering performance)

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

ethosu/vela/tosa_supported_operators.py

diff --git a/ethosu/vela/tosa_supported_operators.py b/ethosu/vela/tosa_supported_operators.py
index 5a85b0e..e378511 100644
--- a/ethosu/vela/tosa_supported_operators.py
+++ b/ethosu/vela/tosa_supported_operators.py
@@ -46,7 +46,7 @@
     activation_ops = relu_ops | set((Op.Table,))
     pad_ops = set((Op.Pad,))
 
-    rank_unlimited_ops = set((Op.Concat, Op.Reshape, Op.Identity))
+    rank_unlimited_ops = set((Op.Concat, Op.Reshape, Op.Identity, Op.Pad))
     rank6_limited_ops = elem_wise_ops
     batch_enabled_ops = rank6_limited_ops | rank_unlimited_ops
     large_tens_dims_enabled_ops = batch_enabled_ops | set((Op.SplitSliceRead,))