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);
父主题: TensorData类