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