设置tensor地址
ge::graphStatus SetAddr(ConstTensorAddressPtr addr, TensorAddrManager manager)
参数 |
输入/输出 |
说明 |
---|---|---|
addr |
输入 |
tensor地址 using ConstTensorAddressPtr = const void *; |
manager |
输入 |
tensor的管理函数 using TensorAddrManager = ge::graphStatus (*)(TensorAddress addr, TensorOperateType operate_type, void **out); |
成功时返回 ge::GRAPH_SUCCESS
失败时返回 manager管理函数中定义的错误码
无。
auto addr = reinterpret_cast<void *>(0x10); TensorData td(addr, nullptr); td.SetAddr(addr, HostAddrManager);