From 9d099ab19dc4887cd4f92ffcd646b4b47a007f9a Mon Sep 17 00:00:00 2001
From: juncaipeng <13006307475@163.com>
Date: Tue, 31 Aug 2021 08:25:01 +0000
Subject: [PATCH] add support ops for quantization

---
 .../paddle/fluid/contrib/slim/quantization/quantization_pass.py  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
index c2d7a9bb4d5174..01b54f8f13a9f1 100644
--- 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']