diff --git a/torch/onnx/ops/__init__.py b/torch/onnx/ops/__init__.py index c0c87d5ccaa..d10ba1ac7a3 100644 --- a/torch/onnx/ops/__init__.py +++ b/torch/onnx/ops/__init__.py @@ -208,7 +208,7 @@ def symbolic_multi_out( # Create a symbolic ONNX operator with the name "CustomOp" in the "custom_domain" domain. # The output tensors will have the specified dtypes and shapes - (out1, out2) = torch.onnx.ops.symbolic( + (out1, out2) = torch.onnx.ops.symbolic_multi_out( "custom_domain::CustomOp", (x,), dict(attr_key="attr_value"),