add support ops for quantization (#35312)

5baccfdd · cc · GitHub · 5c27c2c0 · 5baccfdd
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/fluid/contrib/slim/quantization/quantization_pass.py ...ddle/fluid/contrib/slim/quantization/quantization_pass.py +1 -0

未找到文件。
--- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
@@ -133,6 +133,7 @@ _op_real_in_out_name = {
    "pad2d": [["X"], ["Out"]],
    "flatten": [["X"], ["Out"]],
    "flatten2": [["X"], ["Out"]],
+    "unsqueeze2": [["X"], ["Out"]],
 }

 _conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose']