test

microsoft · Nov 7, 2023 · 41d4812 · 41d4812
1 parent ce973a9
commit 41d4812
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/onnxscript/tests/function_libs/torch_lib/ops_test_data.py b/onnxscript/tests/function_libs/torch_lib/ops_test_data.py
@@ -1947,6 +1947,9 @@ def _where_input_wrangler(
         "ops.aten.tensor.int", core_ops.aten_tensor_int
     ),  # Custom from extra_opinfo
     TorchLibOpInfo("transpose", core_ops.aten_transpose, trace_only=True),
+    TorchLibOpInfo(
+        "transpose", core_ops.aten_transpose_complex, trace_only=True, complex=True
+    ),
     TorchLibOpInfo(
         "var_mean",
         core_ops.aten_var_mean,