Neural network memory allocation method and device and electronic equipment

• Main Author: HUANG CHONGJI
• Format: Patent
• Language: Chinese; English
• Published: 29.09.2023
• Subjects: CALCULATING; COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS; COMPUTING; COUNTING; ELECTRIC DIGITAL DATA PROCESSING; PHYSICS

The embodiment of the invention provides a neural network memory allocation method and device and electronic equipment, and the method comprises the steps: determining a network layer meeting a reusable condition from other network layers in a neural network as a reusable network layer corresponding to a to-be-allocated network layer for each to-be-allocated network layer without memory allocation in the neural network; determining the size relationship between the reusable memory block and the required memory amount of the network layer to be distributed; if the reusable memory block is smaller than the required memory amount, adding a supplementary memory block of which the initial address is adjacent to the end address of the reusable memory block, and allocating the reusable memory block and the supplementary memory block to the network layer to be allocated; and if the reusable network layer does not exist in the allocated network layer, allocating the unallocated memory blocks which are not allocated to