Neural network mapping method and apparatus

A neural network mapping apparatus, comprising a memory, a processor, and a computer program stored in the memory and executable on the processor, wherein the processor, when executing the computer program, implements a neural network mapping method applied to a chip comprising a plurality of processing elements, and wherein the method comprises:

• mapping a calculation task for a preset feature map of each network layer in a plurality of network layers in a convolutional neural network to at least one processing element of the chip;
• acquiring a number of phases needed by the plurality of processing elements in the chip for completing the calculation task, and performing a first stage of balancing on the number of phases of the plurality of processing elements;
• and
• mapping, based on the number of the phases of the plurality of processing elements obtained after the first stage of balancing, the calculation task for the preset feature map of each network layer in the plurality of network layers in the convolutional neural network to at least one processing element of the chip,
• wherein performing the first stage of balancing on the number of phases of the plurality of processing elements comprises:
• determining whether a descent direction of the first stage of balancing exists;
• and
• calculating, in response to a determination result that the descent direction of the first stage of balancing exists, a first-stage balancing reference, and performing the first stage of balancing according to the first-stage balancing reference.

A non-transitory computer-readable storage medium, storing a computer program, wherein the program is executed by a processor to perform a neural network mapping method applied to a chip comprising a plurality of processing elements, comprising:

• mapping a calculation task for a preset feature map of each network layer in a plurality of network layers in a convolutional neural network to at least one processing element of the chip;
• acquiring a number of phases needed by the plurality of processing elements in the chip for completing the calculation task, and performing a first stage of balancing on the number of phases of the plurality of processing elements;
• and
• mapping, based on the number of the phases of the plurality of processing elements obtained after the first stage of balancing, the calculation task for the preset feature map of each network layer in the plurality of network layers in the convolutional neural network to at least one processing element of the chip,
• wherein performing the first stage of balancing on the number of phases of the plurality of processing elements comprises:
• determining whether a descent direction of the first stage of balancing exists;
• and
• calculating, in response to a determination result that the descent direction of the first stage of balancing exists, a first-stage balancing reference, and performing the first stage of balancing according to the first-stage balancing reference.