Documentation

AddAttr

Function Usage

Sets the attributes of an operator and their values.

Prototype

1
2
3
4
5
6
7
8
9
ContextBuilder &AddAttr(const std::string& attrName, int64_t attrValue)
ContextBuilder &AddAttr(const std::string& attrName, bool attrValue)
ContextBuilder &AddAttr(const std::string& attrName, const std::string& attrValue)
ContextBuilder &AddAttr(const std::string& attrName, float attrValue)
ContextBuilder &AddAttr(const std::string& attrName, const std::vector<float>& attrValue)
ContextBuilder &AddAttr(const std::string& attrName, const std::vector<bool>& attrValue)
ContextBuilder &AddAttr(const std::string& attrName, const std::vector<int64_t>& attrValue)
ContextBuilder &AddAttr(const std::string& attrName, const std::vector<std::string>& attrValue)
ContextBuilder &AddAttr(const std::string& attrName, const std::vector<std::vector<int64_t>>& attrValue)

Parameters

Parameter

Input/Output

Description

attrName

Input

Operator attribute name.

attrValue

Input

Operator attribute value. The value can be int64_t, bool, std::string, float, std::vector<float>, std::vector<int64_t>,

std::vector<std::string>, std::vector<bool>, or std::vector<std::vector<int64_t>>.

Returns

Object of the current ContextBuilder

Constraints

AddAttr does not support adding attributes with the same name.

Example

context_ascendc::ContextBuilder builder;
auto builder
    .NodeIoNum(1,1)
    .IrInstanceNum({1})
    .AddAttr("attr_1", 1)
    .AddAttr("attr_2", true)
    .AddAttr("attr_3", "stringValue")
    .AddAttr("attr_4", 1.f)
    .AddAttr("attr_5", {1})
    .AddAttr("attr_6", {false})
    .AddAttr("attr_7", {"stringValue"})
    .AddAttr("attr_8", {1.f})
    .AddAttr("attr_9", {{1, 2}, {3, 4}})
Parent topic: ContextBuilder