函数:set_tensor_place_ment
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
设置Tensor描述的placement属性,用于标识存放Tensor数据的内存为Host内存或Device内存。
函数原型
- C函数原型
1
aclError aclSetTensorPlaceMent(aclTensorDesc *desc, aclMemType memType)
- python函数
1
ret = acl.set_tensor_place_ment(desc, mem_type)
参数说明
参数名 |
说明 |
---|---|
desc |
int,Tensor描述信息的指针地址。 需提前调用acl.create_tensor_desc接口创建算子输入/输出的Tensor描述,作为本接口的输入。 |
mem_type |
int,指定placement属性值,其取值参考aclMemType。 |
返回值说明
返回值 |
说明 |
---|---|
ret |
int,错误码。
|
父主题: aclmdlTensorDesc