conv3d

Author: justinchuby

onnxscript/tests/function_libs/torch_lib/extra_opinfo.py

@@ -69,7 +69,7 @@ def sample_inputs_conv3d(op_info, device, dtype, requires_grad, **kwargs):
             (32,),
             {
                 "stride": (3, 3, 3),
-                "padding": 2,
+                "padding": (2, 2, 2),
                 "dilation": (1, 1, 1),
                 "groups": 1,
             },
@@ -1394,7 +1394,7 @@ def sample_inputs__native_batch_norm_legit_no_stats(
         supports_out=False,
     ),
     opinfo_core.OpInfo(
-        "nn.functional.conv3d",
+        "ops.aten.conv3d",
         aten_name="conv3d",
         dtypes=common_dtype.floating_and_complex_types_and(torch.int64, torch.bfloat16),
         sample_inputs_func=sample_inputs_conv3d,

onnxscript/tests/function_libs/torch_lib/ops_test_data.py

@@ -1879,7 +1879,7 @@ def _where_input_wrangler(
         reason="String padding is not accepted by aten::conv2d",
     ),
     TorchLibOpInfo(
-        "nn.functional.conv3d",
+        "ops.aten.conv3d",
         core_ops.aten_conv3d,
         trace_only=True,
         tolerance={torch.float32: (3.7e-5, 1.8e-4)},