[发明专利]计算图优化方法及装置、电子设备、计算机可读存储介质

权利要求书

1.一种计算图优化方法，其特征在于，包括：

对待处理的第一计算图进行遍历，确定所述第一计算图的第一子图；

将所述第一子图与图模板库中的模板图进行匹配，确定所述模板图中是否存在与第一子图匹配的目标子图；

在存在与所述第一子图匹配的目标子图的情况下，根据所述目标子图，对所述第一计算图进行优化，得到所述第一计算图的优化结果。

2.根据权利要求1所述的方法，其特征在于，所述根据所述目标子图，对所述第一计算图进行优化，得到所述第一计算图的优化结果，包括：

根据所述目标子图的标识，确定与所述目标子图对应的优化方法；

调用与所述目标子图对应的优化方法，对所述第一子图进行优化，得到第二子图；

采用所述第二子图替换所述第一计算图中相应的第一子图，得到优化后的第二计算图；

其中，所述第一计算图的优化结果包括所述第二计算图。

3.根据权利要求1所述的方法，其特征在于，所述根据所述目标子图，对所述第一计算图进行优化，得到所述第一计算图的优化结果，包括：

根据所述目标子图的标识，确定与所述目标子图对应的第一编译流程；

根据与所述目标子图对应的第一编译流程，确定与所述第一计算图对应的整体编译流程；

执行与所述第一计算图对应的整体编译流程，获取所述第一计算图的第一编译结果；

其中，所述第一计算图的优化结果包括所述第一编译结果。

4.根据权利要求3所述的方法，其特征在于，

所述根据与所述目标子图对应的第一编译流程，确定与所述第一计算图对应的整体编译流程，包括：

根据所述目标子图对应的第一子图，确定所述第一编译流程的执行顺序；

根据所述第一计算图中除所述目标子图对应的第一子图外的其他子图，确定与所述第一计算图对应的至少一个第二编译流程以及所述第二编译流程的执行顺序；

所述第一计算图对应的整体编译流程包括所述第一编译流程以及所述第二编译流程；

所述执行与所述第一计算图对应的整体编译流程，获取所述第一计算图的第一编译结果，包括：

按照确定的执行顺序执行所述第一编译流程以及所述第二编译流程，获取所述第一计算图的第一编译结果。

5.根据权利要求4所述的方法，其特征在于，所述根据所述目标子图对应的第一子图，确定所述第一编译流程的执行顺序，包括：

在所述目标子图对应的第一子图为多个的情况下，根据预先设定的所述目标子图对应的第一子图的优先级，确定所述第一编译流程的执行顺序。

6.根据权利要求5所述的方法，其特征在于，所述根据预先设定的所述目标子图对应的第一子图的优先级，确定所述第一编译流程的执行顺序，包括：

在多个所述目标子图对应的第一子图存在交集的情况下，分别确定这多个第一子图对应的优先级；

根据这多个第一子图的优先级顺序，确定这多个第一子图对应的第一编译流程的执行顺序。

7.根据权利要求1-6所述的方法，其特征在于，所述第一计算图为根据神经网络构建的、具有图形式的中间表示；

其中，所述神经网络用于执行处理任务，所述处理任务包括图像处理任务、语音处理任务、文本处理任务、视频处理任务中的任意一种。

8.一种计算图优化装置，其特征在于，包括：

遍历模块，用于对待处理的第一计算图进行遍历，确定所述第一计算图的第一子图；

匹配模块，用于将所述第一子图与图模板库中的模板图进行匹配，确定所述模板图中是否存在与第一子图匹配的目标子图；

优化模块，用于在存在与所述第一子图匹配的目标子图的情况下，根据所述目标子图，对所述第一计算图进行优化，得到所述第一计算图的优化结果。