计算图的图编译方法、装置、设备及存储介质

摘要

本申请公开了一种用于计算图的图编译方法、装置、设备及存储介质，涉及深度学习领域。该方法用于编译深度学习的计算图，该方法包括：获取输入的计算图，计算图包括至少两层节点，相邻两层节点中存在数据流动方向的节点之间连接有有向边，至少两层节点中存在目标节点；对计算图进行图改写，得到优化后的计算图；对于优化后的计算图，生成目标节点的前向计算序列；对前向计算序列中的各个节点进行算子代码编译，得到前向计算序列的第一算子代码；输出目标节点的前向计算序列和第一算子代码。通过对计算图进行改写后，简化了计算图的复杂结构，提升了计算图的编译效率。