[发明专利]计算图的处理方法、装置及设备

说明书

本申请实施例提供一种计算图的处理方法、装置及设备，该方法包括：获取计算图，所述计算图中包含非优化算子；对所述计算图中的所述非优化算子进行转换，得到转换后的优化图，所述优化图中的算子为优化算子。本申请提高了对于计算图的优化能力。

技术领域

本申请涉及计算机技术领域，尤其涉及一种计算图的处理方法、装置及设备。

背景技术

当前，开发人员所开发的算法模型可以抽象成计算图表示，并可以根据推理框架将计算图部署到硬件平台上。

其中，推理框架可以对计算图进行优化，以便计算图在硬件平台可以高效的执行。通常，考虑到算子数量以及硬件平台数量较多，为了减小编写优化代码的成本，推理框架针对常用算子编写不同硬件平台下的优化代码，而针对不常用算子未编写优化代码。由此，在执行计算图时，针对计算图中的常用算子，能够调用常用算子在当前硬件平台下的优化代码，而针对计算图中的不常用算子，只能调用不常用算子的基础代码。

因此，上述方式，存在对计算图的优化能力较弱的问题。

发明内容

本申请实施例提供一种计算图的处理方法、装置及设备，用以解决现有技术中对计算图的优化能力较弱的问题。

第一方面，本申请实施例提供一种计算图的处理方法，包括：

获取计算图，所述计算图中包含非优化算子；

对所述计算图中的所述非优化算子进行转换，得到转换后的优化图，所述优化图中的算子为优化算子。

第二方面，本申请实施例提供一种计算图的处理方法，包括：

获取计算图，所述计算图中包含非优化算子；

对所述计算图中的所述非优化算子进行转换，得到转换后的优化图，所述优化图中的算子为优化算子；

依次调用所述优化图中的算子对应于当前硬件平台的优化代码。

第三方面，本申请实施例提供一种计算图的处理装置，包括：

获取模块，用于获取计算图，所述计算图中包含非优化算子；

转换模块，用于对所述计算图最后那个的所述非优化算子进行转换，得到转换后的优化图，所述优化图中的算子为优化算子。

第四方面，本申请实施例提供一种计算图的处理方法，包括：

获取模块，用于获取计算图，所述计算图中包含非优化算子；

转换模块，用于对所述计算图中的所述非优化算子进行转换，得到转换后的优化图，所述优化图中的算子为优化算子；

调用模块，用于依次调用所述优化图中的算子对应于当前硬件平台的优化代码。

第五方面，本申请实施例提供一种计算设备，包括：存储器、处理器；其中，所述存储器用于存储一条或多条计算机指令，其中，所述一条或多条计算机指令被所述处理器执行时实现如第一方面任一项所述的方法。

第六方面，本申请实施例提供一种计算设备，包括：存储器、处理器；其中，所述存储器用于存储一条或多条计算机指令，其中，所述一条或多条计算机指令被所述处理器执行时实现如第一方面任一项所述的方法。

本申请实施例还提供一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序包含至少一段代码，所述至少一段代码可由计算机执行，以控制所述计算机执行如第一方面任一项所述的方法。

本申请实施例还提供一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序包含至少一段代码，所述至少一段代码可由计算机执行，以控制所述计算机执行如第二方面任一项所述的方法。