diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index 8e5e7751d..197804787 100644 --- a/onnxscript/function_libs/torch_lib/ops/core.py +++ b/onnxscript/function_libs/torch_lib/ops/core.py @@ -8399,7 +8399,7 @@ def aten_unique( @torch_op("aten::_unique2") -def aten_unique2( +def aten__unique2( self: TensorType, sorted: bool = True, return_inverse: bool = False,