GenerateTask
Function Usage
Generates and processes tasks in the GenerateTask phase.
Prototype
1 | OpImplRegisterV2 &GenerateTask(OpGenTaskKernelFunc gen_task_func) |
Parameters
Parameter |
Input/Output |
Description |
||
|---|---|---|---|---|
gen_task_func |
Input |
OpGenTaskKernelFunc function to be registered. The OpGenTaskKernelFunc type is defined as follows:
|
Returns
OpImplRegisterV2 object of the operator, where the OpGenTaskKernelFunc function is registered.
Constraints
None
Parent topic: OpImplRegisterV2