功能说明

定义算子属性。

定义原型

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_;
};

函数说明

表1 OpAttrDef类成员函数说明
函数名称	入参说明	功能说明
AttrType	attr_type: 属性类型	设置算子属性类型，取值为：OPTIONAL（可选）、REQUIRED（必选）。
Bool	无	设置算子属性数据类型为Bool
Bool	value	设置算子属性数据类型为Bool, 并设置属性默认值为value。属性类型设置为OPTIONAL时必须调用该类接口设置默认值。
Float	无	设置算子属性数据类型为Float
Float	value	设置算子属性数据类型为Float, 并设置属性默认值为value。属性类型设置为OPTIONAL时必须调用该类接口设置默认值。
Int	无	设置算子属性数据类型为Int
Int	value	设置算子属性数据类型为Int, 并设置属性默认值为value。属性类型设置为OPTIONAL时必须调用该类接口设置默认值。
String	无	设置算子属性数据类型为String
String	value	设置算子属性数据类型为String, 并设置属性默认值为value。属性类型设置为OPTIONAL时必须调用该类接口设置默认值。
ListBool	无	设置算子属性数据类型为ListBool
ListBool	value	设置算子属性数据类型为ListBool, 并设置属性默认值为value。属性类型设置为OPTIONAL时必须调用该类接口设置默认值。
ListFloat	无	设置算子属性数据类型为ListFloat
ListFloat	value	设置算子属性数据类型为ListFloat, 并设置属性默认值为value。属性类型设置为OPTIONAL时必须调用该类接口设置默认值。
ListInt	无	设置算子属性数据类型为ListInt
ListInt	value	设置算子属性数据类型为ListInt, 并设置属性默认值为value。属性类型设置为OPTIONAL时必须调用该类接口设置默认值。
ListListInt	无	设置算子属性数据类型为ListListInt
ListListInt	value	设置算子属性数据类型为ListListInt, 并设置属性默认值为value。属性类型设置为OPTIONAL时必须调用该类接口设置默认值。
GetName	无	获取属性名称。
IsRequired	无	判断算子属性是否为必选，必选返回true, 可选返回false。
GetCfgDataType	无	该接口预留，开发者无需关注。
GetProtoDataType	无	该接口预留，开发者无需关注。
GetAttrDefaultVal	-	该接口预留，开发者无需关注。

OpAttrDef类

功能说明

定义原型

函数说明