神经网络编译器配置方法、装置、计算机设备和存储介质
基本信息
| 申请号 | CN202111267825.6 | 申请日 | - |
| 公开（公告）号 | CN113703741B | 公开（公告）日 | 2022-02-22 |
| 申请公布号 | CN113703741B | 申请公布日 | 2022-02-22 |
| 分类号 | G06F8/30（2018.01）I;G06N3/063（2006.01）I;G06N3/08（2006.01）I | 分类 | 计算；推算；计数; |
| 发明人 | 白杨;吕江波;沈小勇 | 申请（专利权）人 | 上海思谋科技有限公司 |
| 代理机构 | 广州华进联合专利商标代理有限公司 | 代理人 | 吴平 |
| 地址 | 518051广东省深圳市南山区高新南九道45号三航科技大厦22楼 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本申请涉及一种神经网络编译器配置方法、装置、计算机设备和存储介质。所述方法包括：获取目标神经网络模型，以及目标神经网络模型对应的初始计算图；初始计算图包括多个算子；对多个算子进行划分，得到多个算子集合；确定各算子集合分别对应的多种算子类型组合，并获取各算子类型组合对应的算子运行时间，得到各算子集合分别对应的多个算子运行时间；将多种算子类型组合中，算子运行时间最小的算子类型组合，作为各算子集合对应的目标算子类型组合；根据各算子集合对应的目标算子类型组合生成目标神经网络模型的目标计算图，根据目标计算图生成针对于目标神经网络模型的编译器配置信息。采用本方法能够提高神经网络模型编译器配置优化效果。 |
