From 5f9f6b1714c504b4ac2783915a0ad5814ad0b939 Mon Sep 17 00:00:00 2001 From: a-gardner1 Date: Fri, 17 May 2024 20:48:42 +0000 Subject: [PATCH] Remove unnecessary traceable=True kwargs to torch_op --- onnxscript/function_libs/torch_lib/ops/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index 3acb44ab9..8e5e7751d 100644 --- a/onnxscript/function_libs/torch_lib/ops/core.py +++ b/onnxscript/function_libs/torch_lib/ops/core.py @@ -8398,7 +8398,7 @@ def aten_unique( return result -@torch_op("aten::_unique2", traceable=True) +@torch_op("aten::_unique2") def aten_unique2( self: TensorType, sorted: bool = True, @@ -8417,7 +8417,7 @@ def aten_unique2( return unique_values, inverse_indices, counts -@torch_op("aten::unique_dim", traceable=True) +@torch_op("aten::unique_dim") def aten_unique_dim( self: TensorType, dim: int,