TEBinBuildFn

Function

Build callback function

Syntax

OpRegistrationData& TEBinBuildFn(BuildTeBinFunc buildTeBinFn);

Parameter Description

Parameter	Input/Output	Description
buildTeBinFn	Input	Callback function BuildTeBinFunc. For details, see Callback Function BuildTeBinFunc.

Callback Function BuildTeBinFunc

You can customize and implement the BuildTeBinFunc class function to construct the operator binary file.

Syntax

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

Parameter Description

Parameter	Input/Output	Description
op	Input	Operator data structure of offline model supported by the Ascend AI processor, which stores operator information For details about the operator class, see Operator Class APIs in GE API Reference.
teBinInfo	Output	Path of the binary file of a custom operator and DDK description struct TEBinInfo { std::string bin_file_path; // Automatically obtained from the binFileName field in the JSON file. To ensure compatibility with cases written by users, the field is not deleted. std::string json_file_path; std::string ddk_version; };

Parameter

Input/Output

Description

Input

Operator data structure of offline model supported by the Ascend AI processor, which stores operator information

For details about the operator class, see Operator Class APIs in GE API Reference.

teBinInfo

Output

Path of the binary file of a custom operator and DDK description

struct TEBinInfo

{

std::string bin_file_path; // Automatically obtained from the binFileName field in the JSON file. To ensure compatibility with cases written by users, the field is not deleted.

std::string json_file_path;

std::string ddk_version;

};

Parent topic: Function Reference for Operator Plug-In Development

Previous topic: InferShapeAndTypeFn

Next topic: WeightFormats