arange

microsoft · Jul 22, 2024 · 9451bef · 9451bef
1 parent 7cfae02
commit 9451bef
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -550,9 +550,6 @@ def aten_arange(
 ) -> TensorType:
     """arange(Scalar end, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor"""
 
-    # NOTE: trace_only because both if branches need to be the same type, but we have
-    # a cast in the if branch.
-
     if dtype == -1:
         zero = op.CastLike(0.0, end)
         one = op.CastLike(1.0, end)