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