设置Tensor中数据地址的类型。
void SetPlacement(Placement placement)
参数名
输入/输出
描述
placement
输入
Tensor数据中地址的种类。
枚举值定义如下:
enum Placement { kPlacementHost = 0, // host data addr kPlacementDevice = 1, // device data addr };
无。