[发明专利]一种内存复用方法及其相关产品

权利要求书

1.一种可用于神经网络的内存复用方法，其特征在于，所述内存复用方法包括：

接收用于所述神经网络对应计算图中当前算子的输入数据，并将所述输入数据存储到一存储空间；

通过所述当前算子对所述输入数据进行处理，获得对应的输出数据；

判断所述输入数据是否继续用于所述计算图中其他算子的处理，获取第一判断结果；

根据所述第一判断结果对所述存储空间进行复用；

所述第一判断结果为所述输入数据不再继续用于所述计算图中其他算子的处理时，释放所述输入数据占用的存储空间，以使所述存储空间用于存储其他数据；判断输入数据子集是否继续用于所述当前算子的处理，获取第二判断结果；其中，所述输入数据的数据集包括至少两个输入数据子集；所述存储空间允许存储的数据规模小于所述输入数据的数据集的数据规模；

根据所述第二判断结果对所述存储空间进行复用。

2.如权利要求1所述的内存复用方法，其特征在于，所述其他数据是与所述输入数据不相关的数据。

3.如权利要求1所述的内存复用方法，其特征在于，所述第二判断结果为所述输入数据子集不再继续用于所述当前算子的处理时，释放所述输入数据子集占用的存储空间，以使所述输入数据子集占用的存储空间允许用于存储未被所述当前算子进行处理的其他输入数据子集所对应的输出数据。

4.一种可用于神经网络的内存复用方法，其特征在于，所述内存复用方法包括：

接收用于所述神经网络对应计算图中当前算子的输入数据，并将所述输入数据存储到一存储空间；所述输入数据的数据集包括至少两个输入数据子集；所述存储空间允许存储的数据规模小于所述输入数据的数据集的数据规模；

通过所述当前算子对所述输入数据子集进行处理，获得对应的输出数据；

判断所述输入数据子集是否继续用于所述当前算子的处理，获取第二判断结果；

根据所述第二判断结果对所述存储空间进行复用。

5.如权利要求4所述的内存复用方法，其特征在于，所述第二判断结果为所述输入数据子集不再继续用于所述当前算子的处理时，释放所述输入数据子集对应的存储空间，以使输入数据子集对应的存储空间允许用于存储未被所述当前算子进行处理的其他输入数据子集所对应的输出数据。

6.一种计算机可读存储介质,包括存储于其上的计算机可执行指令，所述可执行指令在被处理器执行时实施如权利要求1-3或4-5中任一项所述的内存复用方法。

7.一种可用于神经网络的内存复用装置，其特征在于，所述内存复用装置包括：

接收用于所述神经网络对应计算图中当前算子的输入数据，并将所述输入数据存储到一存储空间的单元；其中，所述输入数据的数据集包括至少两个输入数据子集；所述存储空间允许存储的数据规模小于所述输入数据的数据集的数据规模；

通过所述当前算子对所述输入数据子集进行处理并获得对应的输出数据的单元；

判断所述输入数据子集是否继续用于所述当前算子的处理，获取第二判断结果的单元；和

根据所述第二判断结果对所述存储空间进行复用的单元。

8.一种计算设备，其特征在于，所述计算设备包括：

一个或多个处理器；和

存储装置，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行时，使得所述一个或多个处理器执行如权利要求1-5中任一项所述的内存复用方法。