下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

AddListFloat

函数功能

向list float类型的属性中追加float元素。

函数原型

void AddListFloat(float f)

参数说明

参数

输入/输出

说明

f

输入

需要添加的float元素。

返回值说明

无。

约束说明

无。

调用示例

例如,OpType为A的算子的原型定义如下所示,属性的名称为attr1,类型为ListFloat。

.ATTR接口中属性类型的取值范围可参见接口说明

REG_OP(A)
    .INPUT(x, TensorType({ DT_FLOAT }))
    .OUTPUT(y, TensorType({ DT_FLOAT }))
    .ATTR(attr1, ListFloat, {1.0})
    .OP_END_FACTORY_REG(A)

则算子实现时可使用如下接口在算子A的属性中追加float类型的属性值。

AttrValue *attr = ctx.GetAttr("attr1");
attr->AddListInt(1.0);
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词