From 4ead54949ac71e8295c55dc7005192cedab64620 Mon Sep 17 00:00:00 2001
From: Chang Xu <molixu7@gmail.com>
Date: Wed, 3 Aug 2022 12:20:07 +0800
Subject: [PATCH] Add Reduce Max in Quant (#44825)

---
 python/paddle/fluid/contrib/slim/quantization/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/paddle/fluid/contrib/slim/quantization/utils.py b/python/paddle/fluid/contrib/slim/quantization/utils.py
index 28efcd2d591..e7187018c8d 100644
--- a/python/paddle/fluid/contrib/slim/quantization/utils.py
+++ b/python/paddle/fluid/contrib/slim/quantization/utils.py
@@ -109,6 +109,7 @@ _act_supported_quantizable_op_type = [
     "square",
     "softplus",
     "shuffle_channel",
+    "reduce_max",
 ]
 
 _out_scale_op_list = list(
@@ -213,6 +214,7 @@ _op_real_in_out_name = {
     "square": [["X"], ["Out"]],
     "softplus": [["X"], ["Out"]],
     "shuffle_channel": [["X"], ["Out"]],
+    "reduce_max": [["X"], ["Out"]],
 }
 
 
-- 
GitLab