[Conv2MatMul] default pad handling for convs

fastmachinelearning · Apr 5, 2024 · 6b8b4ec · 6b8b4ec
1 parent c5bd87f
commit 6b8b4ec
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/qonnx/transformation/lower_convs_to_matmul.py b/src/qonnx/transformation/lower_convs_to_matmul.py
@@ -81,7 +81,8 @@ def apply(self, model):
                     auto_pad = auto_pad.s.decode("utf-8")
                     if auto_pad == "NOTSET":
                         # use specified padding
-                        pad = get_by_name(n.attribute, "pads").ints
+                        pad = get_by_name(n.attribute, "pads")
+                        pad = pad.ints if pad is not None else [0, 0, 0, 0]
                     else:
                         pad = auto_pad_to_explicit_padding(
                             auto_pad,
@@ -95,7 +96,8 @@ def apply(self, model):
                         )
                 else:
                     # use specified padding
-                    pad = get_by_name(n.attribute, "pads").ints
+                    pad = get_by_name(n.attribute, "pads")
+                    pad = pad.ints if pad is not None else [0, 0, 0, 0]
 
                 # If len(pad) == 2, assume no padding for other dimension
                 if len(pad) == 2:  # only one dimension should be padded