fix missing qparams in aten::upsample_nearest2d (#7646)

apache · Mar 12, 2021 · 1a50679 · 1a50679
1 parent 9d72bd0
commit 1a50679
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tvm/relay/frontend/qnn_torch.py b/python/tvm/relay/frontend/qnn_torch.py
@@ -353,6 +353,7 @@ def add_input_quant_params_to_op_inputs(graph):
         "quantized::mul": 2,
         "aten::dequantize": 1,
         "aten::mean": 1,
+        "aten::upsample_nearest2d": 1,
         "aten::upsample_bilinear2d": 1,
         "aten::relu_": 1,
         "aten::relu": 1,