函数:set_tensor_const
产品支持情况
|
产品 |
是否支持 |
|---|---|
|
Atlas 350 加速卡 |
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
功能说明
为算子设置constant输入,指定存放输入Tensor数据的内存地址和地址长度。
函数原型
- C函数原型
1aclError aclSetTensorConst(aclTensorDesc *desc, void *dataBuffer, size_t length)
- python函数
1ret = acl.set_tensor_const(desc, dataBuffer, length)
参数说明
|
参数名 |
说明 |
|---|---|
|
desc |
int,aclTensorDesc类型的指针地址。 需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。 |
|
dataBuffer |
int,存放constant输入Tensor数据的内存地址。 此处算子输入Tensor数据的内存必须根据应用运行模式来确定。内存申请接口请参见内存管理。
|
|
length |
int,内存地址的长度,单位是Byte。 |
父主题: aclmdlTensorDesc