commit ef3ebddb6b4fd951046cbe7799721f6f0ed2fc87
author Patrik Gustavsson <patrik.gustavsson@arm.com> Fri Oct 01 11:10:25 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Fri Oct 01 11:03:38 2021 +0000
tree 9019c0cb4bb1a513ee1b4b00932e16031ab7da65
parent 3ff7a4aed60fefababdced93a5aba91409a4050c

TOSA: Add support for Identity operation

Added support for Identity operation.

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

ethosu/vela/tosa_supported_operators.py

diff --git a/ethosu/vela/tosa_supported_operators.py b/ethosu/vela/tosa_supported_operators.py
index 2692c05..5a85b0e 100644
--- a/ethosu/vela/tosa_supported_operators.py
+++ b/ethosu/vela/tosa_supported_operators.py
@@ -46,13 +46,21 @@
     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