Skip to content

Commit af3e27e

Browse files
authored
support scale op quantization (#45911)
1 parent e755c07 commit af3e27e

File tree

1 file changed

+3
-1
lines changed
  • python/paddle/fluid/contrib/slim/quantization

1 file changed

+3
-1
lines changed

python/paddle/fluid/contrib/slim/quantization/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,7 @@
109109
"softplus",
110110
"shuffle_channel",
111111
"reduce_max",
112+
"scale",
112113
]
113114

114115
_out_scale_op_list = list(
@@ -191,7 +192,7 @@
191192
"fill_any_like": [["X"], ["Out"]],
192193
"fill_constant": [[], ["Out"]],
193194
"gelu": [["X"], ["Out"]],
194-
"instance_norm": [["X"], ["Out"]],
195+
"instance_norm": [["X"], ["Y"]],
195196
"lookup_table": [["W", "Ids"], ["Out"]],
196197
"lookup_table_v2": [["W", "Ids"], ["Out"]],
197198
"norm": [["X"], ["Norm"]],
@@ -214,6 +215,7 @@
214215
"softplus": [["X"], ["Out"]],
215216
"shuffle_channel": [["X"], ["Out"]],
216217
"reduce_max": [["X"], ["Out"]],
218+
"scale": [["X"], ["Out"]],
217219
}
218220

219221

0 commit comments

Comments
 (0)