[发明专利]修改计算图

摘要

包括在计算机存储介质上编码的计算机程序的方法、系统和装置，用于修改计算图以包括发送节点和接收节点。通过将发送节点和接收节点插入到每个子图中，可以有效地处理执行计算图的不同子图的操作的独特设备之间的通信。当被执行时，这些发送节点和接收节点所代表的操作可以使得独特设备对能够以自给自足的方式彼此进行通信。这将协同通信的负担从后端移开，这为处理这种计算图表示的系统提供了在设备执行子图的同时执行一个或多个其他过程的机会。

主权项

1.一种计算机实现的方法，包括：获得表示计算图的数据，所述计算图包括多个节点和有向边，其中每个节点表示相应操作，并且其中每个有向边将相应第一节点连接至相应第二节点，所述相应第二节点表示接收由所述相应第一节点表示的操作的输出作为输入；获得识别跨多个设备的所述计算图的分配的数据，其中所述分配将所述计算图中的每个节点指派给所述多个设备中的相应设备；识别在所述计算图中的一个或多个跨设备有向边，其中每个跨设备有向边是将相应第一节点连接到相应第二节点的有向边，所述相应第二节点在所述分配中被指派给与所述相应第一节点不同的设备；对于每个跨设备有向边：在所述计算图中的所述相应第一节点和所述相应第二节点之间插入发送节点，在所述操作图中的所述发送节点和所述相应第二节点之间插入接收节点，以及修改所述分配以将所述发送节点指派给与所述相应第一节点相同的设备以及将所述接收节点指派给与所述相应第二节点相同的设备；以及根据所修改的分配将由所述计算图中的节点表示的操作指派给所述多个设备。