WINOGRAD TRANSFORM CONVOLUTION OPERATIONS FOR NEURAL NETWORKS

摘要

Some example embodiments may involve performing a convolution operation of a neural network based on a Winograd transform. Some example embodiments may involve a device including neural network processing circuitry that is configured to generate, by the neural network processing circuitry, a transformed input feature map by performing a Winograd transform on an input feature map, the transformed input feature map having a matrix form and including a plurality of channels; to perform, by the neural network processing circuitry, element-wise multiplications between a feature vector of the transformed input feature map and a weight vector of a transformed weight kernel obtained based on the Winograd transform; and to add, by the neural network processing circuitry, element-wise multiplication results, the element-wise multiplications being performed channel-by-channel with respect to the feature vector including feature values on a position in the plurality of channels of the transformed input feature map.