diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-10-05 13:53:34 +0200 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-10-06 09:43:56 +0200 |
commit | b4936adf32435a96ca1a89818582f8d2efa366d1 (patch) | |
tree | b7b15246f1d705d5ba1b1c76eb841b412e11555a /ethosu/vela/tosa_supported_operators.py | |
parent | 352607ca99a4a71de11a241227982b9c95dc69b2 (diff) | |
download | ethos-u-vela-b4936adf32435a96ca1a89818582f8d2efa366d1.tar.gz |
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
Diffstat (limited to 'ethosu/vela/tosa_supported_operators.py')
-rw-r--r-- | ethosu/vela/tosa_supported_operators.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tosa_supported_operators.py b/ethosu/vela/tosa_supported_operators.py index 5a85b0eb..e3785113 100644 --- a/ethosu/vela/tosa_supported_operators.py +++ b/ethosu/vela/tosa_supported_operators.py @@ -46,7 +46,7 @@ class TosaSupportedOperators: 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,)) |