一种神经网络计算图的分割方法、装置、设备及存储介质

专利主权项内容

1.一种神经网络计算图的分割方法，其特征在于，所述方法包括：获取待分割的计算图，所述计算图中包括多个节点以及多个节点之间的边，所述节点中包括计算单元，所述边用于指示计算间的依赖关系；根据各所述节点的处理类型为所述计算图中的各个节点标记对应的设备标识，得到标记后的第一计算图，所述设备标识用于指示各所述节点的执行设备；针对所述第一计算图中具有相同设备标识的节点，根据预定义的子网络结构模板以及各相邻节点之间的融合代价进行节点融合，得到融合后的第二计算图；针对所述第二计算图，根据各所述执行设备对应的内存对所述第二计算图进行分割，得到分割后的目标计算图，所述目标计算图中包括多个子计算图；根据各个子计算图对应的设备标识，将各个子计算图部署至对应的分布式设备上进行并行处理。