文档
注册

函数: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,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

注意事项

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词