From afc0b5f84b3bb901e3919a72441a8554522c37a2 Mon Sep 17 00:00:00 2001 From: Yuan Yao Date: Wed, 9 Oct 2024 16:37:00 -0700 Subject: [PATCH] remove all-zero bias in aten_convolution Signed-off-by: Yuan Yao --- onnxscript/function_libs/torch_lib/ops/core.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index 1fc73a220c..395f1fcac9 100644 --- a/onnxscript/function_libs/torch_lib/ops/core.py +++ b/onnxscript/function_libs/torch_lib/ops/core.py @@ -2030,12 +2030,6 @@ def aten_convolution( stride = (stride, stride) strides = list(stride) - if bias is None: - weight_dim_0 = op.Shape(weight, start=0, end=1) - bias_shape = op.Expand(weight_dim_0, op.Constant(value_ints=[1])) - zero = op.CastLike(0.0, input) - bias = op.Expand(zero, bias_shape) - result = _aten_convolution_onnx( input, weight,