[发明专利]计算图的子图模式匹配方法及装置

权利要求书

1.一种计算图的子图模式匹配方法，其特征在于，所述方法包括：

根据子图中各节点与最终输出节点之间的连接关系，确定子图中各节点的所属层级；

根据所述各节点的顺序编号、节点类型以及节点输出关系，确定所述各节点的路径编码；

根据各所属层级中路径编码不同的节点的顺序编号，确定所述子图的结构描述编码，并根据所述结构描述编码进行不同子图间的模式匹配。

2.根据权利要求1所述的计算图的子图模式匹配方法，其特征在于，所述根据子图中各节点与最终输出节点之间的连接关系，确定子图中各节点的所属层级，包括：

对子图中的最终输出节点设定一默认所属层级，其中，所述最终输出节点为所述子图中没有输出张量的节点；

根据所述子图中各节点与所述最终输出节点之间的节点输入关系和所述最终输出节点的默认所属层级，确定所述各节点的所属层级。

3.根据权利要求2所述的计算图的子图模式匹配方法，其特征在于，所述根据所述子图中各节点与所述最终输出节点之间的节点输入关系和所述最终输出节点的默认所属层级，确定所述各节点的所属层级，包括：

根据所述子图中各节点的节点输出链路数量、所述各节点与所述最终输出节点的节点输入链路上途经节点的节点输出链路数量以及所述最终输出节点的默认所属层级，计算得到所述各节点的所属层级。

4.根据权利要求1所述的计算图的子图模式匹配方法，其特征在于，所述根据所述各节点的顺序编号、节点类型以及节点输出关系，确定所述各节点的路径编码，包括：

以子图中的最终输出节点为起始对各节点进行顺序编号；

根据各节点与相邻节点的顺序编号、各节点与相邻节点的节点输出关系以及各节点的节点类型，确定所述各节点的路径编码。

5.根据权利要求1所述的计算图的子图模式匹配方法，其特征在于，所述根据各所属层级中路径编码不同的节点的顺序编号，确定所述子图的结构描述编码，包括：

若同一所属层级中两相邻节点的路径编码相同，则只保留两相邻节点中较前的顺序编号；

根据各所属层级保留的顺序编号组建而得所述子图的结构描述编码。

6.根据权利要求1所述的计算图的子图模式匹配方法，其特征在于，所述根据所述结构描述编码进行不同子图间的模式匹配，还包括：

判断不同子图的结构描述编码是否相同，若是，则判定两子图所属同一子图类别，并对该两子图进行融合归并计算。

7.一种计算图的子图模式匹配装置，其特征在于，包括：

节点层级划分模块，用于根据子图中各节点与最终输出节点之间的连接关系，确定子图中各节点的所属层级；

节点路径编码模块，用于根据所述各节点的顺序编号、节点类型以及节点输出关系，确定所述各节点的路径编码；

子图结构描述模块，用于根据各所属层级中路径编码不同的节点的顺序编号，确定所述子图的结构描述编码，并根据所述结构描述编码进行不同子图间的模式匹配。

8.根据权利要求7所述的计算图的子图模式匹配装置，其特征在于，所述节点层级划分模块包括：

默认层级设定单元，用于对子图中的最终输出节点设定一默认所属层级，其中，所述最终输出节点为所述子图中没有输出张量的节点；

节点层级计算单元，用于根据所述子图中各节点与所述最终输出节点之间的节点输入关系和所述最终输出节点的默认所属层级，确定所述各节点的所属层级。

9.根据权利要求8所述的计算图的子图模式匹配装置，其特征在于，所述节点层级计算单元包括：

输出链路计算子单元，用于根据所述子图中各节点的节点输出链路数量、所述各节点与所述最终输出节点的节点输入链路上途经节点的节点输出链路数量以及所述最终输出节点的默认所属层级，计算得到所述各节点的所属层级。