设置Tensor的数据存放的位置。
graphStatus SetPlacement(const ge::Placement &placement);
参数名 |
输入/输出 |
描述 |
---|---|---|
placement |
输入 |
需设置的数据地址的值。 枚举值定义如下: enum Placement { kPlacementHost = 0, // host data addr kPlacementDevice = 1, // device data addr }; |
参数名 |
类型 |
描述 |
---|---|---|
- |
graphStatus |
设置成功返回GRAPH_SUCCESS,否则,返回GRAPH_FAILED。 |
无。
无。