Set default values for padding in NHWC pool ops

fastmachinelearning · Apr 5, 2024 · 6084f59 · 6084f59
1 parent a318858
commit 6084f59
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/qonnx/custom_op/general/maxpoolnhwc.py b/src/qonnx/custom_op/general/maxpoolnhwc.py
@@ -53,7 +53,7 @@ def get_nodeattr_types(self):
         # https://github.com/onnx/onnx/blob/main/docs/Operators.md#MaxPool
         return {
             "kernel_shape": ("ints", True, []),
-            "pads": ("ints", True, []),
+            "pads": ("ints", False, [0, 0, 0, 0]),
             "strides": ("ints", True, []),
             "ceil_mode": ("i", False, 0),
         }
@@ -128,7 +128,7 @@ def get_nodeattr_types(self):
         # https://github.com/onnx/onnx/blob/main/docs/Operators.md#AveragePool
         return {
             "kernel_shape": ("ints", True, []),
-            "pads": ("ints", True, []),
+            "pads": ("ints", False, [0, 0, 0, 0]),
             "strides": ("ints", True, []),
             "ceil_mode": ("i", False, 0),
         }