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