Processing method and apparatus for single-channel convolution layer, and processing method and apparatus for multi-channel convolution layer

摘要

A processing method and a processing apparatus for a single-channel convolution layer, and a processing method and apparatus for a multi-channel convolution layer are provided. The processing method for a single-channel convolution layer includes: dividing data to be processed into a plurality of data blocks; reading by a plurality of graphic processor thread groups the plurality of data blocks into local memories of the plurality of graphic processor thread groups; reading by the plurality of graphic processor thread groups a plurality of filters into the local memories of the plurality of graphic processor thread groups; and calculating by a plurality of threads in the plurality of graphic processor thread groups convolutions of corresponding data points in the plurality of data blocks and the plurality of filters simultaneously.