Login| Sign Up| Help| Contact|

Patent Searching and Data


Title:
HARDWARE PLATFORM SPECIFIC OPERATOR FUSION IN MACHINE LEARNING
Document Type and Number:
WIPO Patent Application WO/2021/114530
Kind Code:
A1
Abstract:
A method and associated apparatus for generating a neural network computation graph. The method includes receiving, by a compiler, a computation graph representing a neural network. The computation graph includes a plurality of nodes, each node associated with an operator of the neural network. The compiler receives a list of fusion patterns associated with a target hardware execution device, and analyzes the computation graph using the list of fusion patterns. The compiler generates one or more fused operators based on the analysis, each fused operator including at least two operators of the plurality of operators which can be fused. The compiler generates a new computation graph representing the neural network that includes at least a first fused operator of the generated one or more fused operators.

Inventors:
SHAFIQ FARHAN (CA)
TIAN YE (CA)
ELHOUSHI MOSTAFA (CA)
KRAVCHENKO BORIS (CA)
Application Number:
PCT/CN2020/084352
Publication Date:
June 17, 2021
Filing Date:
April 11, 2020
Export Citation:
Click for automatic bibliography generation   Help
Assignee:
HUAWEI TECH CO LTD (CN)
International Classes:
G06N3/04
Foreign References:
CN110321999A2019-10-11
CN108345937A2018-07-31
CN109948795A2019-06-28
CN109284815A2019-01-29
US20190156178A12019-05-23
Download PDF: