diff options
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 51f80ebd..3b0e6b39 100644 --- a/ethosu/vela/tosa_supported_operators.py +++ b/ethosu/vela/tosa_supported_operators.py @@ -29,7 +29,11 @@ class TosaSupportedOperators: # Categorised lists of supported operators convolution_ops = set((Op.Conv2DBias,)) convolution_like_ops = convolution_ops - mac_main_ops = convolution_like_ops + max_pooling_ops = Op.op_set(Op.is_maxpool_op) + avg_pooling_ops = Op.op_set(Op.is_avgpool_op) + pooling_ops = set((Op.ReduceSum,)) | max_pooling_ops | avg_pooling_ops + + mac_main_ops = convolution_like_ops | pooling_ops type_conversion_ops = set((Op.Rescale,)) relu_ops = set((Op.Clamp, Op.ReluN,)) |