REG_DECOMPOSE_PASS
Function Usage
Developers can register the graph modification function in the 1-to-N scenario with the framework. The framework calls the custom graph modification pass at the beginning of build and calls REG_DECOMPOSE_PASS to register the custom pass.
Prototype
1 | REG_DECOMPOSE_PASS(pass_class, decompose_op_types) |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
pass_class |
Input |
Name of a custom pass. |
decompose_op_types |
Input |
Type of the operator to be replaced. |
Returns
None
Constraints
None
Parent topic: Data Structures and APIs