Skip to content

Commit 8917081

Browse files
elefleatrevor-m
authored andcommitted
fix missing qparams in aten::upsample_nearest2d (apache#7646)
1 parent d942a98 commit 8917081

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/tvm/relay/frontend/qnn_torch.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -353,6 +353,7 @@ def add_input_quant_params_to_op_inputs(graph):
353353
"quantized::mul": 2,
354354
"aten::dequantize": 1,
355355
"aten::mean": 1,
356+
"aten::upsample_nearest2d": 1,
356357
"aten::upsample_bilinear2d": 1,
357358
"aten::relu_": 1,
358359
"aten::relu": 1,

0 commit comments

Comments
 (0)