未验证 提交 af3e27ec 编写于 作者: G Guanghua Yu 提交者: GitHub

support scale op quantization (#45911)

上级 e755c07e
......@@ -109,6 +109,7 @@ _act_supported_quantizable_op_type = [
"softplus",
"shuffle_channel",
"reduce_max",
"scale",
]
_out_scale_op_list = list(
......@@ -191,7 +192,7 @@ _op_real_in_out_name = {
"fill_any_like": [["X"], ["Out"]],
"fill_constant": [[], ["Out"]],
"gelu": [["X"], ["Out"]],
"instance_norm": [["X"], ["Out"]],
"instance_norm": [["X"], ["Y"]],
"lookup_table": [["W", "Ids"], ["Out"]],
"lookup_table_v2": [["W", "Ids"], ["Out"]],
"norm": [["X"], ["Norm"]],
......@@ -214,6 +215,7 @@ _op_real_in_out_name = {
"softplus": [["X"], ["Out"]],
"shuffle_channel": [["X"], ["Out"]],
"reduce_max": [["X"], ["Out"]],
"scale": [["X"], ["Out"]],
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册