文档首页> Atlas 300应用（型号 3000）> Framework API参考> 类成员函数> OpRegistrationData类> GetWorkspaceSizeFn函数
更新时间：2021-10-26 GMT+08:00
查看PDF
分享

GetWorkspaceSizeFn函数

函数功能

获取work 空间大小。

函数原型

OpRegistrationData& GetWorkspaceSizeFn(GetWorkspaceSizeFunc getWorkspaceSizeFn);

参数说明

参数

输入/输出

说明

getWorkspaceSizeFn

输入

回调函数GetWorkspaceSizeFunc，参见回调函数GetWorkspaceSizeFunc。

回调函数GetWorkspaceSizeFunc

用户自定义并实现GetWorkspaceSizeFunc类函数，完成算子对应的工作空间大小计算，并输出。

函数原型

Status GetWorkspaceSizeFn(domi::Status (const ge::Operator& op, std::vector<int64_t>& size));

参数说明

参数

输入/输出

说明

op

输入

需要计算工作空间的算子对象。

关于Operator类，请参见《GE API参考》中的“Operator类接口”。

size

输出

工作空间大小，该值需要用户计算并提供。
父主题： OpRegistrationData类
分享：

    相关文档

    相关产品