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