diff --git a/paddle/fluid/operators/flip_op.cc b/paddle/fluid/operators/flip_op.cc index fc17657594b7a88d15bd2d9f184bc1bf71a71bc2..d7ed5fb767cd9ac9ee9e5024ae1f29c9f12abdcd 100644 --- a/paddle/fluid/operators/flip_op.cc +++ b/paddle/fluid/operators/flip_op.cc @@ -13,10 +13,10 @@ See the License for the specific language governing permissions and limitations under the License. */ #include "paddle/fluid/operators/flip_op.h" - #include #include #include +#include "paddle/fluid/framework/op_version_registry.h" namespace paddle { namespace operators { @@ -154,3 +154,12 @@ REGISTER_OP_CPU_KERNEL( ops::FlipKernel, ops::FlipKernel, ops::FlipKernel); + +/* ========================== register checkpoint ===========================*/ +REGISTER_OP_VERSION(flip) + .AddCheckpoint( + R"ROC(Upgrade flip, add new attr [axis] and delete attr [dims].)ROC", + paddle::framework::compatible::OpVersionDesc() + .NewAttr("axis", "The added attr 'axis' doesn't set default value.", + boost::none) + .DeleteAttr("dims", "The attr 'dims' is deleted."));