AddAttr
函数功能
设置算子的属性以及对应值
函数原型
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) |
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
attrName |
输入 |
算子的属性名称 |
attrValue |
输入 |
算子属性值，支持int64_t, bool, std::string, float, std::vector<float>, std::vector<int64_t>,std::vector<std::string>,std::vector<bool>,std::vector<std::vector<int64_t>>类型的指定 |
返回值说明
当前ContextBuilder的对象。
约束说明
AddAttr不支持重复添加同名的属性
调用示例
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}})
父主题： ContextBuilder类