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 2ae9f21 commit 3778f7dCopy full SHA for 3778f7d
onnxscript/function_libs/torch_lib/ops/core.py
@@ -7689,7 +7689,7 @@ def aten_remainder(self: TTensor, other: float) -> TTensor:
7689
7690
7691
@torch_op("aten::remainder.Scalar_Tensor", trace_only=True)
7692
-def aten_remainder_scalar_tensor(self: TTensor, other: TTensor) -> TTensor:
+def aten_remainder_scalar_tensor(self: float, other: TTensor) -> TTensor:
7693
"""remainder.Scalar_Tensor(Scalar self, Tensor other) -> Tensor"""
7694
7695
self_tensor = ir.tensor(self, dtype=other.dtype)
0 commit comments