Function Description
Sets the value of an operator attribute.
An operator may have multiple attributes. Once the value of an attribute is initially set, the data type of the attribute is fixed. The following data types are supported:
Integer: int64_t, uint32_t, and int32_t
Use SetAttr(const string& name, int64_t attrValue) to set the attribute value, and use GetAttr(const string& name, int32_t& attrValue) or GetAttr(const string& name, uint32_t& attrValue) to obtain the value. Ensure that the integer data is not clipped. It is the same for the types int32_t and uint32_t.
Integer list: std::vector[/topic/body/section/ul/li/p/b/int64_t {""}) (int64_t], std::vector[/topic/body/section/ul/li/p/b/int32_t {""}) (int32_t], std::vector[/topic/body/section/ul/li/p/b/uint32_t {""}) (uint32_t], or std::initializer_list[/topic/body/section/ul/li/p/b/int64_t {""}) && (int64_t]
Floating-point number: float
Floating-point number list: std::vector<float>
String
String list: std::vector<float>
bool
Bool list: std::vector<bool>
Tensor: Tensor
Tensor list: std::vector<Tensor>
Arrays of bytes: OpBytes (that is, vector<uint8_t>) or (const uint8_t* data, size_t size)
AttrValue: AttrValue
2D integer list: std::vector<std::vector<int64_t>>
DataType list: std::vector<ge::DataType>
DataType: ge::DataType
NamedAttrs: ge::NamedAttrs
NamedAttrs list: std::vector<ge::NamedAttrs>
Function Prototype
APIs taking string arguments will be deprecated in future releases. Use APIs taking non-string arguments instead.
Operator &SetAttr(const std::string &name, int64_t attr_value); Operator &SetAttr(const std::string &name, int32_t attr_value); Operator &SetAttr(const std::string &name, uint32_t attr_value); Operator &SetAttr(const std::string &name, const std::vector<int64_t> &attr_value); Operator &SetAttr(const std::string &name, const std::vector<int32_t> &attr_value); Operator &SetAttr(const std::string &name, const std::vector<uint32_t> &attr_value); Operator &SetAttr(const std::string &name, std::initializer_list<int64_t> &&attr_value); Operator &SetAttr(const std::string &name, float32_t attr_value); Operator &SetAttr(const std::string &name, const std::vector<float32_t> &attr_value); Operator &SetAttr(const std::string &name, AttrValue &&attr_value); Operator &SetAttr(const std::string &name, const std::string &attr_value); Operator &SetAttr(const std::string &name, const std::vector<std::string> &attr_value); Operator &SetAttr(const std::string &name, bool attr_value); Operator &SetAttr(const std::string &name, const std::vector<bool> &attr_value); Operator &SetAttr(const std::string &name, const Tensor &attr_value); Operator &SetAttr(const std::string &name, const std::vector<Tensor> &attr_value); Operator &SetAttr(const std::string &name, const OpBytes &attr_value); Operator &SetAttr(const std::string &name, const std::vector<std::vector<int64_t>> &attr_value); Operator &SetAttr(const std::string &name, const std::vector<ge::DataType> &attr_value); Operator &SetAttr(const std::string &name, const ge::DataType &attr_value); Operator &SetAttr(const std::string &name, const ge::NamedAttrs &attr_value); Operator &SetAttr(const std::string &name, const std::vector<ge::NamedAttrs> &attr_value); Operator &SetAttr(const char_t *name, int64_t attr_value); Operator &SetAttr(const char_t *name, int32_t attr_value); Operator &SetAttr(const char_t *name, uint32_t attr_value); Operator &SetAttr(const char_t *name, const std::vector<int64_t> &attr_value); Operator &SetAttr(const char_t *name, const std::vector<int32_t> &attr_value); Operator &SetAttr(const char_t *name, const std::vector<uint32_t> &attr_value); Operator &SetAttr(const char_t *name, std::initializer_list<int64_t> &&attr_value); Operator &SetAttr(const char_t *name, float32_t attr_value); Operator &SetAttr(const char_t *name, const std::vector<float32_t> &attr_value); Operator &SetAttr(const char_t *name, AttrValue &&attr_value); Operator &SetAttr(const char_t *name, const char_t *attr_value); Operator &SetAttr(const char_t *name, const AscendString &attr_value); Operator &SetAttr(const char_t *name, const std::vector<AscendString> &attr_values); Operator &SetAttr(const char_t *name, bool attr_value); Operator &SetAttr(const char_t *name, const std::vector<bool> &attr_value); Operator &SetAttr(const char_t *name, const Tensor &attr_value); Operator &SetAttr(const char_t *name, const std::vector<Tensor> &attr_value); Operator &SetAttr(const char_t *name, const OpBytes &attr_value); Operator &SetAttr(const char_t *name, const std::vector<std::vector<int64_t>> &attr_value); Operator &SetAttr(const char_t *name, const std::vector<ge::DataType> &attr_value); Operator &SetAttr(const char_t *name, const ge::DataType &attr_value); Operator &SetAttr(const char_t *name, const ge::NamedAttrs &attr_value); Operator &SetAttr(const char_t *name, const std::vector<ge::NamedAttrs> &attr_value);
Parameters
| Parameter | Input/Output | Description |
|---|---|---|
| name | Input | Attribute name. |
| attr_value | Input | Attribute value of type int64_t. |
| attr_value | Input | Attribute value of type int32_t. |
| attr_value | Input | Attribute value of type uint32_t. |
| attr_value | Input | Attribute value of type vector<int64_t>. |
| attr_value | Input | Attribute value of type vector<int32_t>. |
| attr_value | Input | Attribute value of type vector<uint32_t>. |
| attr_value | Input | Attribute value of type std::initializer_list[/topic/body/section/table/tgroup/tbody/row/entry/b/int64_t {""}) && (int64_t]. |
| attr_value | Input | Attribute value of the floating-point type. |
| attr_value | Input | Attribute value of the floating-point list type. |
| attr_value | Input | Attribute value of the bool type. |
| attr_value | Input | Attribute value of the bool list type. |
| attr_value | Input | Attribute value of type AttrValue. |
| attr_value | Input | Attribute value of the string type. |
| attr_value | Input | Attribute value of the string list type. |
| attr_value | Input | Attribute value of type Tensor. |
| attr_value | Input | Attribute value of the tensor list type. |
| attr_value | Input | Attribute value of type OpBytes, or vector[/topic/body/section/table/tgroup/tbody/row/entry/b/uint8_t {""}) (uint8_t]. |
| attr_value | Input | Attribute value of the quantized data type. |
| attr_value | Input | Attribute value of type vector<vector[/topic/body/section/table/tgroup/tbody/row/entry/b/int64_t {""}) > (int64_t]. |
| attr_value | Input | Attribute value of type vectorge::DataType. |
| attr_value | Input | Attribute value of type DataType. |
| attr_value | Input | Attribute value of type NamedAttrs. |
| attr_value | Input | Attribute value of type vectorge::NamedAttrs. |