ParseParamsByOperatorFn
Description
Registers a function for parsing the attributes of a custom operator.
Prototype
OpRegistrationData &ParseParamsByOperatorFn(const ParseParamByOpFunc &parse_param_by_op_fn)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
parse_param_by_op_fn |
Input |
Function for parsing the attributes of a custom operator. For details, see Callback Function ParseParamByOpFunc. |
Callback Function ParseParamByOpFunc
Call the user-defined ParseParamByOpFunc function to map operator attributes of the original model to those of the model adapted to Ascend AI Processor and fill the result in class Operator.
Status ParseParamByOpFunc(const ge::Operator &op_origin, ge::Operator &op_dest)
Parameter |
Input/Output |
Description |
|---|---|---|
op_origin |
Input |
An object of class Operator defined by the framework, including the custom operator attributes of the original model. For details about Operator class, see Operator. |
op_dest |
Output |
Operator data structure of the model adapted to Ascend AI Processor, which stores operator information. For details about class Operator, see Operator. |
Restrictions
None