下载
中文
注册

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