机器学习模型的编译优化方法和装置
摘要:
本发明公开了一种机器学习模型的编译优化方法和装置。其中,该方法包括:多次运行第一机器学习模型对一组预热数据进行处理,采集处理过程的模型运行信息;根据所述模型运行信息,确定所述第一组操作中的张量形状未发生变化的操作作为目标操作;在所述目标操作包括所述第一组操作中的多个操作的情况下,通过所述目标编译器对所述第一机器学习模型中的所述目标操作进行合并,得到第二机器学习模型。本发明解决了由于在机器学习在编译时没有引入运行时的采样信息,导致的编译周期长、浪费资源的技术问题。
公开/授权文献
0/0