shanyi15-patch-3
tangwei12 7 years ago
parent eff92d046b
commit cd98f2b7e0

@ -154,8 +154,7 @@ class CheckpointLoadOp : public framework::OperatorBase {
class CheckpointLoadOpProtoMaker : public framework::OpProtoAndCheckerMaker { class CheckpointLoadOpProtoMaker : public framework::OpProtoAndCheckerMaker {
public: public:
CheckpointLoadOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker) void Make() override {
: OpProtoAndCheckerMaker(proto, op_checker) {
AddInput( AddInput(
"X", "X",
"(vector) Input LoDTensors that need to be saved together in a file.") "(vector) Input LoDTensors that need to be saved together in a file.")

@ -143,8 +143,7 @@ class CheckpointSaveOp : public framework::OperatorBase {
class CheckpointSaveOpProtoMaker : public framework::OpProtoAndCheckerMaker { class CheckpointSaveOpProtoMaker : public framework::OpProtoAndCheckerMaker {
public: public:
CheckpointSaveOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker) void Make() override {
: OpProtoAndCheckerMaker(proto, op_checker) {
AddInput( AddInput(
"X", "X",
"(vector) Input LoDTensors that need to be saved together in a file.") "(vector) Input LoDTensors that need to be saved together in a file.")

Loading…
Cancel
Save