SetFp16Data

函数功能

针对通过AllocHostTensor申请得到的host侧tensor，用一块float16类型的内存初始化tensor数据。

函数原型

void SetFp16Data(const op::fp16_t *value, uint64_t size, op::DataType dataType)

参数说明

返回值说明

无

约束说明

入参指针不能为空。

调用示例

1
2
3
4
5

// 初始化一块fp16内存，赋值给input的前10个元素
void Func(const aclTensor *input) {
    fp16_t myArray[10];
    input->SetData(myArray, 10, DT_FLOAT16);
}