CNIPA.AI
검색으로 돌아가기
기록

计算图优化方法和装置、电子设备及存储介质

발명유효
17청구항 · 4 독립항
§ Ⅰ

개요

발명자

请求不公布姓名; 请求不公布姓名; 请求不公布姓名

IPC 분류

G06F 9/50 (2006.01)G06N 3/063 (2023.01)G06F 17/16 (2006.01)

本公开的实施例提供一种计算图优化方法和装置、电子设备及存储介质。该方法包括:获取第一计算图,其中,每个第一节点对应的第一算子预配置有适配不同类型的硬件计算核心的至少两个候选核函数;根据第一算子的属性信息,从第一算子对应的核函数容器列表中确定目标核函数容器;根据每个第一算子对应的目标核函数容器和属性信息,确定是否对第一算子对应的第一节点执行局部图优化操作;针对确定执行局部图优化操作的待优化节点,根据对应的目标核函数容器执行局部图优化操作以得到第二计算图;根据第二计算图,生成第一计算图对应的可执行代码。该方法可以在泛化性与极致性能间取得最优平衡,避免引入冗余计算开销,具有较强灵活性。