跳到主要内容

SetPlacement

函数功能​

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

函数原型​

void SetPlacement(Placement placement);

参数说明​

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

返回值​

无

异常处理​

无

约束说明​

无