diff options
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 3685c5ae..998d94ff 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -615,7 +615,7 @@ class Operation: is_supported = False if self.original_type == Op.ResizeBilinear and self.type == Op.DepthwiseConv2DBias: is_supported = True - if self.original_type == Op.AvgPool and self.type == Op.DepthwiseConv2DBias: + if self.original_type == Op.AvgPool and self.type in (Op.DepthwiseConv2DBias, Op.Conv2DBias): is_supported = True if is_supported: |