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

trevor-m · May 6, 2021 · f340d74 · f340d74
1 parent da5ea55
commit f340d74
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,