OpAttrDef类

功能说明

定义算子属性。

定义原型

class OpAttrDef {
public:
  explicit OpAttrDef(const char *name);
  OpAttrDef(const OpAttrDef &attr_def);
  ~OpAttrDef();
  OpAttrDef &operator=(const OpAttrDef &attr_def);
  bool operator==(const OpAttrDef &attr_def) const;
  OpAttrDef &AttrType(Option attr_type);
  OpAttrDef &Bool(void);
  OpAttrDef &Bool(bool value);
  OpAttrDef &Float(void);
  OpAttrDef &Float(float value);
  OpAttrDef &Int(void);
  OpAttrDef &Int(int64_t value);
  OpAttrDef &String(void);
  OpAttrDef &String(const char *value);
  OpAttrDef &ListBool(void);
  OpAttrDef &ListBool(std::vector<bool> value);
  OpAttrDef &ListFloat(void);
  OpAttrDef &ListFloat(std::vector<float> value);
  OpAttrDef &ListInt(void);
  OpAttrDef &ListInt(std::vector<int64_t> value);
  OpAttrDef &ListListInt(void);
  OpAttrDef &ListListInt(std::vector<std::vector<int64_t>> value);
  ge::AscendString &GetName(void) const;
  bool IsRequired(void);
  ge::AscendString &GetCfgDataType(void) const;
  ge::AscendString &GetProtoDataType(void) const;
  ge::AscendString &GetAttrDefaultVal(const char *brac);
private:
  std::unique_ptr<OpAttrDefImpl> impl_;
};

函数说明