Method and electronic device for convolution calculation in neutral network
AssigneeNanjing Horizon Robotics Technology Co., Ltd.
InventorsLiang CHEN, Chang HUANG, Kun LING, Jianjun LI, Delin LI, Heng LUO
Life of this patent
Estimating the term…
Loading patent…
AssigneeNanjing Horizon Robotics Technology Co., Ltd.
InventorsLiang CHEN, Chang HUANG, Kun LING, Jianjun LI, Delin LI, Heng LUO
Estimating the term…
Disclosed is a method for convolution calculation in a neural network, comprising: reading an input feature map, depthwise convolution kernels and pointwise convolution kernels from a dynamitic random access memory (DRAM); performing depthwise convolution calculations and pointwise convolution calculations according to the input feature map, the depthwise convolution kernels and the pointwise convolution kernels to obtain output feature values of a first predetermined number p of points on all pointwise convolution output channels; storing the output feature values of a first predetermined number p of points on all pointwise convolution output channels into an on-chip memory, wherein the first predetermined number p is determined according to at least one of available space in the on-chip memory, a number of the depthwise convolution calculation units, and width, height and channel dimensions of the input feature map; and repeating the above operation obtain output feature values of all points on all pointwise convolution output channels. Therefore, the storage space for storing intermediate results may be reduced.
A method for convolution calculation in a neural network, comprising:
An electronic device, comprising:
Dependencies are inferred from the claim text.