add support ops for quantization

juncaipeng · juncaipeng · commit 9d099ab19dc4 · 2021-08-31T08:25:01.000Z
diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
@@ -133,6 +133,7 @@
     "pad2d": [["X"], ["Out"]],
     "flatten": [["X"], ["Out"]],
     "flatten2": [["X"], ["Out"]],
+    "unsqueeze2": [["X"], ["Out"]],
 }
 
 _conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose']

Original file line number	Diff line number	Diff line change
`@@ -133,6 +133,7 @@`
`133`	`133`	`"pad2d": [["X"], ["Out"]],`
`134`	`134`	`"flatten": [["X"], ["Out"]],`
`135`	`135`	`"flatten2": [["X"], ["Out"]],`
	`136`	`+ "unsqueeze2": [["X"], ["Out"]],`
`136`	`137`	`}`
`137`	`138`
`138`	`139`	`_conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose']`