diff options
Diffstat (limited to 'ethosu/vela/supported_operators.py')
-rw-r--r-- | ethosu/vela/supported_operators.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index fb3061f7..1e11d788 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -142,7 +142,7 @@ class SupportedOperators: def check_convolution_restrictions(self, op): # check stride - if op.attrs["stride_w"] > 2 or op.attrs["stride_h"] > 2: + if op.attrs["stride_w"] > 3 or op.attrs["stride_h"] > 3: return False # check dilation @@ -182,7 +182,7 @@ class SupportedOperators: def check_pooling_restrictions(self, op): # check stride - if op.attrs["stride_w"] > 2 or op.attrs["stride_h"] > 2: + if op.attrs["stride_w"] > 3 or op.attrs["stride_h"] > 3: return False # check data type |