diff --git a/onnxscript/function_libs/torch_aten/ops/core.py b/onnxscript/function_libs/torch_aten/ops/core.py index b1cb886a36..de48ee8a80 100644 --- a/onnxscript/function_libs/torch_aten/ops/core.py +++ b/onnxscript/function_libs/torch_aten/ops/core.py @@ -1984,7 +1984,7 @@ def aten_gcd(self: TensorType, other: TensorType) -> TensorType: def aten_ge(self: TReal, other: TReal) -> BOOL: # ge.Tensor(Tensor self, Tensor other) -> Tensor - return op.Greater(self, other) + return op.GreaterOrEqual(self, other) def aten_geqrf(self: TensorType) -> tuple[TensorType, TensorType]: @@ -2542,7 +2542,7 @@ def aten_ldexp(self: TensorType, other: TensorType) -> TensorType: def aten_le(self: TReal, other: TReal) -> BOOL: # le.Tensor(Tensor self, Tensor other) -> Tensor - return op.Less(self, other) + return op.LessOrEqual(self, other) def aten_lerp(self: TensorType, end: TensorType, weight: TensorType) -> TensorType: