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"]], }