aclSetTensorConst
函数功能
为算子设置constant输入,指定存放输入tensor数据的内存地址和地址长度。
函数原型
aclError aclSetTensorConst(aclTensorDesc *desc, void *dataBuffer, size_t length)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
desc |
输出 |
aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
dataBuffer |
输入 |
输入tensor数据内存地址指针。 此处算子输入tensor数据的内存必须根据应用运行模式来确定,应用运行在Host时,此处需申请Host上的内存;应用运行在Device时,此处需申请Device上的内存。内存申请接口请参见内存管理。 |
length |
输入 |
内存地址的长度,单位是Byte。 |
返回值说明
返回0表示成功,返回其他值表示失败。
父主题: aclTensorDesc