diff options
| author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-08-25 13:49:25 +0200 |
|---|---|---|
| committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-03 14:37:43 +0200 |
| commit | b081d67f3e96896ea0ca7426a18c5fe153419ee0 (patch) | |
| tree | a62a36256836b9bebcf318fe71793ad800099267 /ethosu/vela/tosa_supported_operators.py | |
| parent | df99510f04aef99d1b8e9be9bfcde8fc1738b65f (diff) | |
| download | ethos-u-vela-b081d67f3e96896ea0ca7426a18c5fe153419ee0.tar.gz | |
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
Diffstat (limited to 'ethosu/vela/tosa_supported_operators.py')
| -rw-r--r-- | ethosu/vela/tosa_supported_operators.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/ethosu/vela/tosa_supported_operators.py b/ethosu/vela/tosa_supported_operators.py index 90d54687..d7a1ebc2 100644 --- a/ethosu/vela/tosa_supported_operators.py +++ b/ethosu/vela/tosa_supported_operators.py @@ -40,12 +40,16 @@ class TosaSupportedOperators: 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 |