diff --git a/onnxscript/tests/function_libs/torch_lib/ops_test_data.py b/onnxscript/tests/function_libs/torch_lib/ops_test_data.py index 7af316b36..c5aeb1668 100644 --- 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,