commit b081d67f3e96896ea0ca7426a18c5fe153419ee0
author Patrik Gustavsson <patrik.gustavsson@arm.com> Wed Aug 25 13:49:25 2021 +0200
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Fri Sep 03 14:37:43 2021 +0200
tree a62a36256836b9bebcf318fe71793ad800099267
parent df99510f04aef99d1b8e9be9bfcde8fc1738b65f

TOSA: Added support for ADD, SUB and MUL

Added support for ADD, SUB and MUL

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

ethosu/vela/tosa_supported_operators.py

diff --git a/ethosu/vela/tosa_supported_operators.py b/ethosu/vela/tosa_supported_operators.py
index 90d5468..d7a1ebc 100644
--- a/ethosu/vela/tosa_supported_operators.py
+++ b/ethosu/vela/tosa_supported_operators.py
@@ -40,12 +40,16 @@
     mac_main_ops = convolution_like_ops | pooling_ops | fc_vector_products
     memory_only_ops = set((Op.Reshape, Op.Transpose,))
 
+    binary_elem_wise_add_mul_sub = set((Op.Add, Op.Mul, Op.RescaleMul, Op.Sub,))
+
     type_conversion_ops = set((Op.Rescale,))
     relu_ops = set((Op.Clamp, Op.ReluN,))
     activation_ops = relu_ops
 
     npu_post_ops = activation_ops
-    supported_operators = mac_main_ops | type_conversion_ops | npu_post_ops | memory_only_ops
+    supported_operators = (
+        mac_main_ops | type_conversion_ops | npu_post_ops | memory_only_ops | binary_elem_wise_add_mul_sub
+    )
 
     # Supported data types
     # TODO will differ compared to TensorFlow Lite, currently set to the same