[发明专利]调度计算图上的操作

权利要求书

1.一种方法，包括：

由计算图系统接收生成用于处理计算图的调度的请求，其中，所述调度定义由所述计算图表示的根据排序来执行的操作序列；

获得表示所述计算图的数据，所述计算图包括多个节点和有向边，其中，每个节点表示相应操作，

其中，从相应第一节点到相应第二节点的每个有向边表示由所述相应第二节点表示的第二操作需要通过执行由所述相应第一节点表示的第一操作而生成的至少一个输出作为输入；

生成所述计算图的分隔器，其中，所述分隔器满足从所述计算图中移除所述分隔器中的节点和将其他节点连接到所述分隔器的边使得所述计算图的剩余节点和边形成多个连接分量的特性；以及

生成所述调度以执行所述计算图中表示的操作，其中，生成所述调度包括：

利用零节点来初始化所述调度；

对于所述分隔器中的每个节点：

确定所述节点在所述计算图中是否具有任何前趋节点，

当所述节点具有任何前趋节点时，将所述前趋节点添加到所述调度，以及

将所述节点添加在所述调度中，以及

将在所述计算图上既不在所述分隔器中也不是所述分隔器中的任何节点的前趋的每个子图中的每个节点添加到所述调度。

2.根据前述权利要求中的任一项所述的方法，还包括根据所述调度来处理所述计算图。

3.根据前述权利要求中的任一项所述的方法，其中，所述计算图中表示的每个操作是与处理神经网络的输入相关联的操作。

4.根据前述权利要求中的任一项所述的方法，其中，生成所述分隔器包括生成用于所述计算图的树分解。

5.根据前述权利要求中的任一项所述的方法，其中，所述分隔器是所述树分解的平衡分隔器，并且其中，生成用于所述计算图的所述树分解包括将所述树分解生成为小于预定宽度。

6.根据任一项前述权利要求所述的方法，其中，所述调度包括注释，每个注释与所述调度中的相应节点相对应并且指示通过执行与所述相应节点相对应的相应操作而生成的输出是否被保存在存储器中。

7.一种方法，包括：

由计算图系统接收生成用于处理计算图的调度的请求，其中，所述调度定义由所述计算图表示的根据排序来执行的操作序列；

获得表示所述计算图的数据，所述计算图包括多个节点和有向边，

其中，每个节点表示相应操作，以及

其中，从相应第一节点到相应第二节点的每个有向边表示由所述相应第二节点表示的第二操作需要通过执行由所述相应第一节点表示的第一操作而生成的至少一个输出作为输入；

确定所述计算图是否包括遍历所述计算图中的每个节点的通过所述计算图的第一有向路径，包括通过特定节点；

响应于确定所述计算图包括所述第一有向路径：

生成路径分解，所述路径分解定义(i)多个集合，所述多个集合中的每个集合包括所述计算图的一个或多个节点，以及(ii)通过所述多个集合中的每个集合的第二有向路径，

利用零节点来初始化所述调度，

标识所述路径分解的所述多个集合中的包括所述特定节点的特定集合，

对于所述特定集合中的每个节点并且按照根据所述第一有向路径的顺序：

确定所述节点在所述计算图中是否具有任何前趋节点

当所述节点在所述计算图中具有任何前趋节点时，将所述前趋节点添加到所述调度，以及

在所述前趋节点之后将所述节点添加到所述调度。