commit 1a92f78e14f31f1423824228deb0628b7a9a9071
author Louis Verhaard <louis.verhaard@arm.com> Tue Feb 09 16:08:26 2021 +0100
committer Louis Verhaard <louis.verhaard@arm.com> Wed Feb 17 16:29:20 2021 +0100
tree 4e0de9a2e6b5b7d9159b25cbfead4a625c134a3c
parent 8d0f4890aa0ceae92a33ebb789701ff644a6fcaa

MLBEDSW-4022: support PAD followed by pool operator

PAD followed by max/average pool is run on NPU if NPU
padding can be used. Average pool is converted to depthwise.

Change-Id: Icc3652e6d9ecff5ac3dc7d92080313d90c245404
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>

ethosu/vela/high_level_command_to_npu_op.py

diff --git a/ethosu/vela/high_level_command_to_npu_op.py b/ethosu/vela/high_level_command_to_npu_op.py
index b5e7b4b..1059e6e 100644
--- a/ethosu/vela/high_level_command_to_npu_op.py
+++ b/ethosu/vela/high_level_command_to_npu_op.py
@@ -133,7 +133,8 @@
         and op.kernel.elements_wh() == 1
     ):
         rounding_mode = NpuRoundingMode.NATURAL
-    rounding_mode = op.attrs.get("rounding_mode", rounding_mode)
+    if op.rounding_mode is not None:
+        rounding_mode = op.rounding_mode
     return rounding_mode