Update onnxscript/function_libs/torch_lib/ops/core.py

Co-authored-by: Justin Chu <[email protected]>
microsoft · May 20, 2024 · 4a39cb6 · 4a39cb6
1 parent 5f9f6b1
commit 4a39cb6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,