diff --git a/python/paddle/fluid/contrib/slim/quantization/utils.py b/python/paddle/fluid/contrib/slim/quantization/utils.py index 28efcd2d5915745bed05c54102ca2956b2caf1e4..e7187018c8d5bffb8c0b5bb3872093159e1cdbe7 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"]], }