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

support scale op quantization (#45911)

上级 e755c07e
...@@ -109,6 +109,7 @@ _act_supported_quantizable_op_type = [ ...@@ -109,6 +109,7 @@ _act_supported_quantizable_op_type = [
"softplus", "softplus",
"shuffle_channel", "shuffle_channel",
"reduce_max", "reduce_max",
"scale",
] ]
_out_scale_op_list = list( _out_scale_op_list = list(
...@@ -191,7 +192,7 @@ _op_real_in_out_name = { ...@@ -191,7 +192,7 @@ _op_real_in_out_name = {
"fill_any_like": [["X"], ["Out"]], "fill_any_like": [["X"], ["Out"]],
"fill_constant": [[], ["Out"]], "fill_constant": [[], ["Out"]],
"gelu": [["X"], ["Out"]], "gelu": [["X"], ["Out"]],
"instance_norm": [["X"], ["Out"]], "instance_norm": [["X"], ["Y"]],
"lookup_table": [["W", "Ids"], ["Out"]], "lookup_table": [["W", "Ids"], ["Out"]],
"lookup_table_v2": [["W", "Ids"], ["Out"]], "lookup_table_v2": [["W", "Ids"], ["Out"]],
"norm": [["X"], ["Norm"]], "norm": [["X"], ["Norm"]],
...@@ -214,6 +215,7 @@ _op_real_in_out_name = { ...@@ -214,6 +215,7 @@ _op_real_in_out_name = {
"softplus": [["X"], ["Out"]], "softplus": [["X"], ["Out"]],
"shuffle_channel": [["X"], ["Out"]], "shuffle_channel": [["X"], ["Out"]],
"reduce_max": [["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.
先完成此消息的编辑!
想要评论请 注册