更新时间:2021-10-26 GMT+08:00
分享

GetGetWorkspaceSizeFunc函数

函数功能

获取workspace大小回调函数。

函数原型

domi::GetWorkspaceSizeFunc GetGetWorkspaceSizeFunc(const std::string& op_type);

参数说明

参数

输入/输出

说明

op_type

输入

算子类型

分享:

    相关文档

    相关产品