From fb5d31c3f42a2a5357822b6072792ce62447cff1 Mon Sep 17 00:00:00 2001 From: shubhambhokare1 Date: Thu, 25 Jul 2024 23:17:56 +0000 Subject: [PATCH] fix unused vars --- onnxscript/function_libs/torch_lib/ops/nn.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/onnxscript/function_libs/torch_lib/ops/nn.py b/onnxscript/function_libs/torch_lib/ops/nn.py index 10a3e2db85..a0d04aa426 100644 --- a/onnxscript/function_libs/torch_lib/ops/nn.py +++ b/onnxscript/function_libs/torch_lib/ops/nn.py @@ -690,7 +690,6 @@ def _get_im2col_indices_along_dim( return block_mask - def _get_im2col_padded_input(input, padding_h, padding_w): # Input is always 4-D tensor (N, C, H, W) # Padding tensor has the following format: (padding_h, padding_w) @@ -747,7 +746,7 @@ def aten_im2col( stride = (stride, stride) strides = list(stride) - stride_h, stride_w = stride[0], stride[1] + stride_h, stride_w = strides[0], strides[1] padding_h, padding_w = pads[0], pads[1] dilation_h, dilation_w = dilations[0], dilations[1] kernel_h, kernel_w = kernel_sizes[0], kernel_sizes[1]