更新时间:2021-03-18 GMT+08:00
分享

TEBinBuildFn函数

函数功能

build回调函数。

函数原型

OpRegistrationData& TEBinBuildFn(BuildTeBinFunc buildTeBinFn);

参数说明

参数

输入/输出

说明

buildTeBinFn

输入

回调函数BuildTeBinFunc,参见回调函数BuildTeBinFunc

回调函数BuildTeBinFunc

用户自定义并实现BuildTeBinFunc类函数,用于构建算子二进制文件。

函数原型

virtual Status BuildTeBinFunc(const ge::Operator& op,TEBinInfo& teBinInfo);

参数说明

参数

输入/输出

说明

op

输入

适配昇腾AI处理器的离线模型的算子数据结构,保存算子信息。

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

teBinInfo

输出

存储自定义算子二进制文件路径和ddk描述的数据。

struct TEBinInfo

{

std::string bin_file_path; //自动从json文件binFileName字段获取。为了兼容以前用户编写的用例,字段不删除。

std::string json_file_path;

std::string ddk_version;

};

分享:

    相关文档

    相关产品