diff options
author | Raul Farkas <raul.farkas@arm.com> | 2023-05-09 09:09:17 +0100 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2023-06-16 12:26:19 +0000 |
commit | 3e7157ba59f12aa0d277a9b3a7cb3f8a19267338 (patch) | |
tree | a06a40060fc1d9a44b3688fea916b61d26c56a65 /ethosu/vela/operation.py | |
parent | 3b64f068db4ea8e954a1b472de169dd423b8c049 (diff) | |
download | ethos-u-vela-3e7157ba59f12aa0d277a9b3a7cb3f8a19267338.tar.gz |
MLBEDSW-7315: Add support for AvgPool with stride_width > 3
* Convert AvgPool with stride_width > 3 and Valid padding to Conv2D to
optimize it to run on NPU.
Change-Id: I06ab412357f0b09b1498f9019a9d1963a324ad34
Signed-off-by: Raul Farkas <raul.farkas@arm.com>
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: |