You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2021/03/12 06:22:26 UTC
[tvm] branch main updated: fix missing qparams in
aten::upsample_nearest2d (#7646)
This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 1a50679 fix missing qparams in aten::upsample_nearest2d (#7646)
1a50679 is described below
commit 1a506791ba7b7f041de318b6a83c2a7f080ee1cd
Author: eleflea <dm...@gmail.com>
AuthorDate: Fri Mar 12 14:22:04 2021 +0800
fix missing qparams in aten::upsample_nearest2d (#7646)
---
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 2b85a1f..2dd84b6 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,