Prototype Registration API (OP_ADD)
Function
Registers the prototype definition of an operator to ensure that the operator can be correctly identified, built, and executed by the framework.
The operator prototype describes the input, output, and attributes of the operator as well as the implementation information of the operator on the AI Processor, and associates the operator with functions such as tiling implementation. The operator prototype is carried by a custom operator class that is inherited from the OpDef class After defining the operator prototype, call the OP_ADD API to pass in the operator type (class name of the custom operator class) and register the operator prototype. For details, see Operator Prototype Definition.
Prototype
1
|
OP_ADD(opType) |
Parameters
|
Parameter |
Input/Output |
Description |
|---|---|---|
|
opType |
Input |
Operator type name. |
Returns
None
Constraints
None