diff --git a/paddle/fluid/operators/checkpoint_load_op.cc b/paddle/fluid/operators/checkpoint_load_op.cc index a9676de369b4b45a097835ad104e37f745742ec5..82a40e18d5c9e5748eac8eb454257e8598d6bdc4 100644 --- a/paddle/fluid/operators/checkpoint_load_op.cc +++ b/paddle/fluid/operators/checkpoint_load_op.cc @@ -154,8 +154,7 @@ class CheckpointLoadOp : public framework::OperatorBase { class CheckpointLoadOpProtoMaker : public framework::OpProtoAndCheckerMaker { public: - CheckpointLoadOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker) - : OpProtoAndCheckerMaker(proto, op_checker) { + void Make() override { AddInput( "X", "(vector) Input LoDTensors that need to be saved together in a file.") diff --git a/paddle/fluid/operators/checkpoint_save_op.cc b/paddle/fluid/operators/checkpoint_save_op.cc index 30eda30c5f52fb3eea6c58015400a65161de3cf9..790fd4ea68505a306872290cadb71319038ad965 100644 --- a/paddle/fluid/operators/checkpoint_save_op.cc +++ b/paddle/fluid/operators/checkpoint_save_op.cc @@ -143,8 +143,7 @@ class CheckpointSaveOp : public framework::OperatorBase { class CheckpointSaveOpProtoMaker : public framework::OpProtoAndCheckerMaker { public: - CheckpointSaveOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker) - : OpProtoAndCheckerMaker(proto, op_checker) { + void Make() override { AddInput( "X", "(vector) Input LoDTensors that need to be saved together in a file.")