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