From bc18551371d33ea6c6c8519ff8b789cef59d6203 Mon Sep 17 00:00:00 2001
From: Jay Tsou <hsjts0u@gmail.com>
Date: Thu, 4 Sep 2025 14:01:23 -0700
Subject: [PATCH] Add default args for _aten_con2d

---
 torchax/torchax/ops/jaten.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/torchax/torchax/ops/jaten.py b/torchax/torchax/ops/jaten.py
index 851a2d6103e..700d581d773 100644
--- a/torchax/torchax/ops/jaten.py
+++ b/torchax/torchax/ops/jaten.py
@@ -1017,11 +1017,11 @@ def _aten_bucketize(input,
 def _aten_conv2d(
     input,
     weight,
-    bias,
-    stride,
-    padding,
-    dilation,
-    groups,
+    bias=None,
+    stride=[1, 1],
+    padding=[0, 0],
+    dilation=[1, 1],
+    groups=1,
 ):
   return _aten_convolution(
       input,