SetPlacement
函数功能
设置Tensor的数据存放的位置。
函数原型
void SetPlacement(Placement placement);
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
placement |
输入 |
需设置的数据地址的值。 枚举值定义如下： enum Placement {
kPlacementHost = 0, // host data addr
kPlacementDevice = 1, // device data addr
}; |
返回值
无。
异常处理
无。
约束说明
无。
父主题： TensorDesc类