Skip to content

Commit

Permalink
Remove unnecessary traceable=True kwargs to torch_op
Browse files Browse the repository at this point in the history
  • Loading branch information
a-gardner1 committed May 20, 2024
1 parent 48467a2 commit 5f9f6b1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions onnxscript/function_libs/torch_lib/ops/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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,
Expand Down

0 comments on commit 5f9f6b1

Please sign in to comment.