diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-10-01 11:10:25 +0200 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-10-01 11:03:38 +0000 |
commit | ef3ebddb6b4fd951046cbe7799721f6f0ed2fc87 (patch) | |
tree | 9019c0cb4bb1a513ee1b4b00932e16031ab7da65 /ethosu/vela/tosa_supported_operators.py | |
parent | 3ff7a4aed60fefababdced93a5aba91409a4050c (diff) | |
download | ethos-u-vela-ef3ebddb6b4fd951046cbe7799721f6f0ed2fc87.tar.gz |
TOSA: Add support for Identity operation
Added support for Identity operation.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: If00b30528932f7531807ce3914d6c1875ab72fa4
Diffstat (limited to 'ethosu/vela/tosa_supported_operators.py')
-rw-r--r-- | ethosu/vela/tosa_supported_operators.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/ethosu/vela/tosa_supported_operators.py b/ethosu/vela/tosa_supported_operators.py index 2692c05f..5a85b0eb 100644 --- a/ethosu/vela/tosa_supported_operators.py +++ b/ethosu/vela/tosa_supported_operators.py @@ -46,13 +46,21 @@ class TosaSupportedOperators: activation_ops = relu_ops | set((Op.Table,)) pad_ops = set((Op.Pad,)) - rank_unlimited_ops = set((Op.Concat, Op.Reshape)) + rank_unlimited_ops = set((Op.Concat, Op.Reshape, Op.Identity)) 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,)) npu_post_ops = activation_ops - supported_operators = mac_main_ops | type_conversion_ops | npu_post_ops | memory_only_ops | elem_wise_ops | pad_ops + supported_operators = ( + mac_main_ops + | type_conversion_ops + | npu_post_ops + | memory_only_ops + | elem_wise_ops + | pad_ops + | set((Op.Identity,)) + ) # Supported data types # TODO will differ compared to TensorFlow Lite, currently set to the same |