[发明专利]地址分配方法及装置

权利要求书

1.一种地址分配方法，其特征在于，所述方法包括：

在对神经网络进行编译时，根据神经网络的计算图生成针对神经网络的每一层的维度转换指令；

根据输入数据的维度以及神经网络的每一层对应的维度转换指令确定每一层的输出数据的维度；

针对每一层的输出数据，根据该层的输出数据的维度为该层的输出数据分配第一内存空间得到该层的输出数据的存储地址，记录该层的输出数据的存储地址；

所述计算图包括神经网络各层的算子，根据神经网络的计算图生成针对神经网络的每一层的维度转换指令，包括：

针对神经网络的每一层，根据该层的算子生成该层对应的维度转换指令；

根据输入数据的维度以及神经网络的每一层对应的维度转换指令确定每一层的输出数据的维度，包括：

从神经网络的第一层到最后一层，针对神经网络的每一层，根据该层对应的维度转换指令对该层的输入数据进行维度转换以得到该层的输出数据的维度，

其中，该层的输入数据是该层的上一层的输出数据，该层的输出数据是该层的下一层的输入数据。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

根据神经网络的层数分配第二内存空间，所述第二内存空间用于存储神经网络的每一层的输出数据的存储地址；

获取第二内存空间的基地址、以及神经网络的每一层的输出数据的存储地址的存储位置相对于所述基地址的偏移量。

3.根据权利要求2所述的方法，其特征在于，记录该层的输出数据的存储地址，包括：

根据该层的输出数据的存储地址的存储位置相对于所述基地址的偏移量将该层的输出数据的存储地址记录在所述第二内存空间中。

4.根据权利要求1所述的方法，其特征在于，对该层的输入数据进行维度转换包括：对该层的输入数据的H维度和W维度进行维度转换。

5.根据权利要求1所述的方法，其特征在于，所述算子包括用于生成维度转换指令的函数。

6.一种地址分配装置，其特征在于，所述装置包括：

指令生成模块，用于在对神经网络进行编译时，根据神经网络的计算图生成针对神经网络的每一层的维度转换指令；

确定模块，用于根据输入数据的维度以及神经网络的每一层对应的维度转换指令确定每一层的输出数据的维度；

第一内存分配模块，用于针对每一层的输出数据，根据该层的输出数据的维度为该层的输出数据分配第一内存空间得到该层的输出数据的存储地址，记录该层的输出数据的存储地址；

所述计算图包括神经网络各层的算子，指令生成模块，包括：

指令生成单元，用于针对神经网络的每一层，根据该层的算子生成该层对应的维度转换指令；

所述确定模块包括：

维度转换单元，用于从神经网络的第一层到最后一层，针对神经网络的每一层，根据该层对应的维度转换指令对该层的输入数据进行维度转换以得到该层的输出数据的维度，

其中，该层的输入数据是该层的上一层的输出数据，该层的输出数据是该层的下一层的输入数据。

7.根据权利要求6所述的装置，其特征在于，所述装置还包括：

第二内存分配模块，用于根据神经网络的层数分配第二内存空间，所述第二内存空间用于存储神经网络的每一层的输出数据的存储地址；

获取模块，用于获取第二内存空间的基地址、以及神经网络的每一层的输出数据的存储地址的存储位置相对于所述基地址的偏移量。

8.根据权利要求7所述的装置，其特征在于，所述第一内存分配模块包括：

记录单元，用于根据该层的输出数据的存储地址的存储位置相对于所述基地址的偏移量将该层的输出数据的存储地址记录在所述第二内存空间中。

9.根据权利要求6所述的装置，其特征在于，对该层的输入数据进行维度转换包括：对该层的输入数据的H维度和W维度进行维度转换。