跳到主要内容

SetPlacement

函数功能

设置Tensor的数据存放的位置。

函数原型

void SetPlacement(Placement placement);

参数说明

参数名输入/输出描述
placement输入需设置的数据地址的值。 枚举值如下:kPlacementHost、kPlacementDevice。

返回值

异常处理

约束说明