From d199243d8907b2d8062dd9c20b69dcb9765a970f Mon Sep 17 00:00:00 2001 From: eleflea Date: Fri, 12 Mar 2021 10:04:07 +0800 Subject: [PATCH] fix missing qparams in aten::upsample_nearest2d --- python/tvm/relay/frontend/qnn_torch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/relay/frontend/qnn_torch.py b/python/tvm/relay/frontend/qnn_torch.py index 2b85a1f3a1be..2dd84b650bd2 100644 --- 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,