We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 735f955 commit ece0819Copy full SHA for ece0819
onnxscript/function_libs/torch_lib/ops/core.py
@@ -1239,7 +1239,7 @@ def aten_bitwise_and(self: TTensor, other: TTensor) -> TTensor:
1239
1240
1241
@torch_op("aten::bitwise_and.Scalar", trace_only=True)
1242
-def aten_bitwise_and_scalar(self: TTensor, other: float) -> TTensor:
+def aten_bitwise_and_scalar(self: TTensor, other: int) -> TTensor:
1243
"""bitwise_and.Scalar(Tensor self, Scalar other) -> Tensor"""
1244
1245
other_tensor = op.Constant(value=ir.tensor(other, dtype=self.dtype))
0 commit comments