From c030589df65081dcc95f5366c8db84990537d7ad Mon Sep 17 00:00:00 2001 From: Yang Yang Date: Fri, 20 Oct 2017 22:52:51 +0000 Subject: [PATCH] add optional in op proto --- paddle/framework/framework.proto | 1 + paddle/framework/op_proto_maker.h | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/paddle/framework/framework.proto b/paddle/framework/framework.proto index 2aa961f140..3d023535ef 100644 --- a/paddle/framework/framework.proto +++ b/paddle/framework/framework.proto @@ -68,6 +68,7 @@ message OpProto { optional bool duplicable = 3 [ default = false ]; optional bool intermediate = 4 [ default = false ]; + optional bool dispensable = 5 [ default = false ]; } // AttrProto describes the C++ type Attribute. diff --git a/paddle/framework/op_proto_maker.h b/paddle/framework/op_proto_maker.h index a134befd90..44e8ab1689 100644 --- a/paddle/framework/op_proto_maker.h +++ b/paddle/framework/op_proto_maker.h @@ -44,6 +44,11 @@ class OpProtoAndCheckerMaker { var_->set_intermediate(true); return *this; } + + VariableBuilder& AsDispensable() { + var_->set_dispensable(true); + return *this; + } }; VariableBuilder AddInput(const std::string& name, const std::string& comment); -- GitLab