Skip to content

Commit 0197eee

Browse files
BowenBaogchanan
authored andcommitted
rename scales
1 parent 89554d7 commit 0197eee

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/onnx/symbolic_caffe2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ def dequantize(g, input):
138138
def _empty_affine_quantized(g, input, shape, scale, zero_point, dtype, pin_memory, memory_format, layout):
139139
return input
140140

141-
def upsample_nearest2d(g, input, output_size, align_corners=None, scales_1=-1, scales_2=-1.0):
141+
def upsample_nearest2d(g, input, output_size, align_corners=None, scales_h=-1, scales_w=-1.0):
142142
if input not in sym_help._quantized_ops:
143143
from torch.onnx.symbolic_opset9 import upsample_nearest2d as upsample_nearest2d_impl
144144
return upsample_nearest2d_impl(g, input, output_size, align_corners)

0 commit comments

Comments
 (0)