Invention Grant
- Patent Title: Method and electronic device for convolution calculation in neural network
-
Application No.: US16222693Application Date: 2018-12-17
-
Publication No.: US11574031B2Publication Date: 2023-02-07
- Inventor: Liang Chen , Chang Huang , Kun Ling , Jianjun Li , Delin Li , Heng Luo
- Applicant: Nanjing Horizon Robotics Technology Co., Ltd.
- Applicant Address: CN Nanjing
- Assignee: Nanjing Horizon Robotics Technology Co., Ltd.
- Current Assignee: Nanjing Horizon Robotics Technology Co., Ltd.
- Current Assignee Address: CN Nanjing
- Agency: K&L Gates LLP
- Main IPC: G06F17/15
- IPC: G06F17/15 ; G06N3/063 ; G06N3/04 ; G06N3/10

Abstract:
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.
Public/Granted literature
- US20190197083A1 METHOD AND ELECTRONIC DEVICE FOR CONVOLUTION CALCULATION IN NEUTRAL NETWORK Public/Granted day:2019-06-27
Information query