Title:
NEURAL NETWORK OPERATION METHOD AND APPARATUS, CHIP, ELECTRONIC DEVICE AND STORAGE MEDIUM
Document Type and Number:
WIPO Patent Application WO/2023/098256
Kind Code:
A1
Abstract:
The present application relates to a neural network operation method and apparatus, a chip, an electronic device and a storage medium. The neural network operation method comprises: acquiring input data and Wk*Hk sub-convolution kernel groups for neural network operation, and executing an operation step, N Wk*Hk*C convolution kernels of the neural network operation being split into N*Wk*Hk 1*1*C sub-convolution kernels, and the N*Wk*Hk 1*1*C sub-convolution kernels being divided into Wk*Hk sub-convolution kernel groups. The operation step comprises: rearranging the input data on the basis of a data rearrangement mode corresponding to each sub-convolution kernel group so as to obtain rearranged input data corresponding to each sub-convolution kernel group; and convolving each sub-convolution kernel group with the rearranged input data corresponding to the sub-convolution kernel group to obtain a convolution result of the sub-convolution kernel group; accumulating the convolution results of the sub-convolution kernel groups to obtain an accumulation result, and taking data located at a valid position in the accumulation result as an output result of the neural network operation.
Inventors:
XU DONG (CN)
XIONG XIANKUI (CN)
XIONG XIANKUI (CN)
Application Number:
PCT/CN2022/121427
Publication Date:
June 08, 2023
Filing Date:
September 26, 2022
Export Citation:
Assignee:
ZTE CORP (CN)
International Classes:
G06N3/04; G06N3/08
Foreign References:
| US20190188237A1 | 2019-06-20 | |||
| US20190065896A1 | 2019-02-28 | |||
| CN112215745A | 2021-01-12 | |||
| CN108241890A | 2018-07-03 | |||
| CN111260037A | 2020-06-09 |
Attorney, Agent or Firm:
CENFO INTELLECTUAL PROPERTY AGENCY (CN)
Download PDF: