SetListString
函数功能
设置list string类型的属性值。
函数原型
void SetListString(const std::vector<std::string> &bytes)
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
bytes |
输入 |
需要设置的list string类型的属性值。 |
返回值说明
无。
约束说明
无。
调用示例
例如,OpType为A的算子的原型定义如下所示,属性的名称为attr1,类型为ListString。
.ATTR接口中属性类型的取值范围可参见接口说明。
REG_OP(A) .INPUT(x, TensorType({ DT_FLOAT })) .OUTPUT(y, TensorType({ DT_FLOAT })) .ATTR(attr1, ListString, {}) .OP_END_FACTORY_REG(A)
则算子实现时可使用如下接口设置算子A的属性值。
AttrValue *attr = ctx.GetAttr("attr1"); std::vector<std::string> strs; strs.push_back("seed1"); strs.push_back("seed2"); attr->SetListString(strs);
父主题: AttrValue类