设置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);