Add HasAttr method to Operator

4e54ab76 · Michal Gallus · 700bcbf7 · 4e54ab76
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/framework/operator.h paddle/fluid/framework/operator.h +1 -0

未找到文件。
--- 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",