Scheduling method for a multi-layer convolutional neural network, electronic device and storage medium

Abstract

A scheduling method, a scheduling apparatus, an electronic device and a storage medium are provided. The scheduling method includes: respectively performing, by a plurality of compute units, a first convolution computation on a plurality of data groups that correspond to the plurality of compute units respectively, to obtain a plurality of first computation result groups; determining a data replication-transmission mode corresponding to the plurality of first computation result groups in the plurality of compute units, according to a configuration rule of a second convolutional layer in the plurality of compute units; and copying a first intermediate data row required by the first compute unit for padding in the second convolution computation process from a first computation result group in the second compute unit and transmitting the first intermediate data row that is copied to the first compute unit based on the data replication-transmission mode.

CPC Classifications