From 9451bef7bdfed0ad9bfe12f519b0cc298a485649 Mon Sep 17 00:00:00 2001 From: Justin Chu Date: Mon, 22 Jul 2024 23:17:59 +0000 Subject: [PATCH] arange --- onnxscript/function_libs/torch_lib/ops/core.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index de7e0a88f..306d060c9 100644 --- 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)