support scale op quantization (#45911)

PaddlePaddle · Sep 9, 2022 · af3e27e · af3e27e
1 parent e755c07
commit af3e27e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/paddle/fluid/contrib/slim/quantization/utils.py b/python/paddle/fluid/contrib/slim/quantization/utils.py
@@ -109,6 +109,7 @@
     "softplus",
     "shuffle_channel",
     "reduce_max",
+    "scale",
 ]
 
 _out_scale_op_list = list(
@@ -191,7 +192,7 @@
     "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 @@
     "softplus": [["X"], ["Out"]],
     "shuffle_channel": [["X"], ["Out"]],
     "reduce_max": [["X"], ["Out"]],
+    "scale": [["X"], ["Out"]],
 }