设置Tensor的数据存放的位置。
1
void SetPlacement(Placement placement)
参数名
输入/输出
描述
placement
输入
需设置的数据地址的值。
枚举值定义如下：
1 2 3 4
enum Placement { kPlacementHost = 0, // host data addr kPlacementDevice = 1, // device data addr };
无。