From b23faf37be2c1c396fee329cdbbef29a74195bf4 Mon Sep 17 00:00:00 2001 From: guofei <52460041+gfwm2013@users.noreply.github.com> Date: Tue, 29 Dec 2020 19:17:58 +0800 Subject: [PATCH] Add moving_average_abs_max_scale op_register_version test=develop (#29957) Add moving_average_abs_max_scale op_register_version --- paddle/fluid/operators/fake_quantize_op.cc | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/paddle/fluid/operators/fake_quantize_op.cc b/paddle/fluid/operators/fake_quantize_op.cc index df4debb620..abfc88e515 100644 --- a/paddle/fluid/operators/fake_quantize_op.cc +++ b/paddle/fluid/operators/fake_quantize_op.cc @@ -813,3 +813,11 @@ REGISTER_OP_VERSION(fake_channel_wise_quantize_abs_max) "quantization to conv2d_tranpose and mul ops.)ROC", paddle::framework::compatible::OpVersionDesc().NewAttr( "quant_axis", "The axis for quantization.", 0)); +REGISTER_OP_VERSION(moving_average_abs_max_scale) + .AddCheckpoint( + R"ROC(Incompatible upgrade of output [Out])ROC", + paddle::framework::compatible::OpVersionDesc().DeleteOutput( + "Out", + "Delete output in order to make the inference model not " + "save moving_average_abs_max_scale operator. This will " + "make the quantitative model be correctly applied in inference.")); -- GitLab