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