From defbb203765c6d184b1fab648149563ec401ed03 Mon Sep 17 00:00:00 2001 From: RachelXu7 Date: Tue, 2 Aug 2022 10:56:00 +0000 Subject: [PATCH] Add Reduce Max in Quant --- 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 28efcd2d59157..e7187018c8d5b 100644 --- a/python/paddle/fluid/contrib/slim/quantization/utils.py +++ b/python/paddle/fluid/contrib/slim/quantization/utils.py @@ -109,6 +109,7 @@ "square", "softplus", "shuffle_channel", + "reduce_max", ] _out_scale_op_list = list( @@ -213,6 +214,7 @@ "square": [["X"], ["Out"]], "softplus": [["X"], ["Out"]], "shuffle_channel": [["X"], ["Out"]], + "reduce_max": [["X"], ["Out"]], }