From ef3ebddb6b4fd951046cbe7799721f6f0ed2fc87 Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Fri, 1 Oct 2021 11:10:25 +0200 Subject: TOSA: Add support for Identity operation Added support for Identity operation. Signed-off-by: Patrik Gustavsson Change-Id: If00b30528932f7531807ce3914d6c1875ab72fa4 --- ethosu/vela/tosa_supported_operators.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'ethosu/vela/tosa_supported_operators.py') 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 -- cgit v1.2.1