Merge pull request #4990 from tonyyang-svail/add_optional_in_op_proto

add optional in op proto
PaddlePaddle · Oct 21, 2017 · fa05094 · fa05094
2 parents ac4f759 + c030589
commit fa05094
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git 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
@@ -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);