CANN/metadef GenerateTask接口
GenerateTask【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef函数功能GenerateTask阶段具体Task的生成和处理。函数原型OpImplRegisterV2 GenerateTask(OpGenTaskKernelFunc gen_task_func)参数说明参数输入/输出说明gen_task_func输入待注册的OpGenTaskKernelFunc函数。OpGenTaskKernelFunc类型定义如下using OpGenTaskKernelFunc UINT32 (*)(const ExeResGenerationContext *context, std::vectorstd::vectoruint8_t tasks);返回值说明返回算子的OpImplRegisterV2对象本身该对象新增注册OpGenTaskKernelFunc函数。约束说明无【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2598374.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!