GenerateTask

GenerateTask阶段具体Task的生成和处理。

OpImplRegisterV2 &GenerateTask(OpGenTaskKernelFunc gen_task_func)

参数

输入/输出

说明

gen_task_func

输入

待注册的OpGenTaskKernelFunc函数。

OpGenTaskKernelFunc类型定义如下：

using OpGenTaskKernelFunc = UINT32 (*)(const ExeResGenerationContext *context, std::vector<std::vector<uint8_t>> &tasks);

返回算子的OpImplRegisterV2对象本身，该对象新增注册OpGenTaskKernelFunc函数。

无

父主题： OpImplRegisterV2