SetPlacement
函数功能
设置tensor的placement。
函数原型
1 | void SetPlacement(const TensorPlacement placement) |
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
placement |
输入 |
tensor的placement。 关于TensorPlacement类型的定义，请参见TensorPlacement。 |
返回值说明
无。
约束说明
无。
调用示例
1 2 3 | auto addr = reinterpret_cast<void *>(0x10); TensorData td(addr, nullptr); auto td_place = td.SetPlacement(kOnHost); |
父主题： TensorData