Skip to content

Commit 3505420

Browse files
authored
[torchlib] Add back operator and/or (#2590)
Previously the entries were mistakenly removed. --------- Signed-off-by: Justin Chu <[email protected]>
1 parent 3a26097 commit 3505420

File tree

1 file changed

+3
-1
lines changed
  • onnxscript/function_libs/torch_lib/ops

1 file changed

+3
-1
lines changed

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1239,6 +1239,7 @@ def aten_binomial(
12391239
"aten::bitwise_and.Tensor",
12401240
"aten::bitwise_and.Scalar",
12411241
"aten::bitwise_and.Scalar_Tensor",
1242+
"_operator::and_",
12421243
),
12431244
trace_only=True,
12441245
)
@@ -1354,6 +1355,7 @@ def aten_bitwise_not(self: TTensor) -> TTensor:
13541355
"aten::bitwise_or.Tensor",
13551356
"aten::bitwise_or.Scalar",
13561357
"aten::bitwise_or.Scalar_Tensor",
1358+
"_operator::or_",
13571359
),
13581360
trace_only=True,
13591361
)
@@ -5051,7 +5053,7 @@ def aten_logical_not(self: TTensor) -> BOOL:
50515053
return op.Not(op.Cast(self, to=BOOL.dtype))
50525054

50535055

5054-
@torch_op(("aten::logical_or"), trace_only=True)
5056+
@torch_op("aten::logical_or", trace_only=True)
50555057
def aten_logical_or(self: TTensor, other: TTensor) -> BOOL:
50565058
"""logical_or(Tensor self, Tensor other) -> Tensor"""
50575059

0 commit comments

Comments
 (0)