摘要
本公开提供了一种深度学习模型的编译方法、装置、设备、介质和程序产品,涉及人工智能技术领域,尤其涉及深度学习等技术领域。深度学习模型的编译方法的具体实现方案为:基于针对深度学习模型的初始中间表示计算图的优化代码,生成指示待替换子图的第一计算图和指示替换后子图的第二计算图;对第一计算图与初始中间表示计算图进行匹配;以及响应于匹配到初始中间表示计算图中包括与第一计算图匹配的待替换子图,基于第二计算图,对初始中间表示计算图中的待替换子图进行替换,得到优化后中间表示计算图,以基于优化后中间表示计算图生成处理器进行深度学习推理的执行代码,其中,优化代码为基于通用语言编写的代码。
技术关键词
深度学习模型
节点
编译方法
神经网络芯片
搜索算法
处理器
子模块
编译装置
匹配模块
指令
关系
模式
接口
人工智能技术
计算机程序产品
显卡
校验模块
电子设备