Add HasAttr method to Operator

7 years ago · 4e54ab76ec
parent 700bcbf74f
commit 4e54ab76ec
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/framework/operator.h
+++ b/paddle/fluid/framework/operator.h
@ -100,6 +100,7 @@ class OperatorBase {

  const std::string& Type() const { return type_; }

+  bool HasAttr(const std::string& name) const { return attrs_.count(name); }
  template <typename T>
  inline const T& Attr(const std::string& name) const {
    PADDLE_ENFORCE(attrs_.count(name) != 0, "%s should be in AttributeMap",