提交 cd98f2b7 编写于 作者: T tangwei12

bug fix

上级 eff92d04
......@@ -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.")
......
......@@ -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.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册