设置TensorDesc的placement属性,用于标识存放Tensor数据的内存类型。同步接口。
aclError aclSetTensorPlaceMent(aclTensorDesc *desc, aclMemType memType)
参数名 |
输入/输出 |
说明 |
---|---|---|
desc |
输出 |
aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
memType |
输入 |
指定placement属性值。 |
返回0表示成功,返回其它值表示失败。