Operator fusion method and device, equipment and storage medium
The invention provides an operator fusion method and device, equipment and a storage medium, and relates to the technical field of artificial intelligence, in particular to the technical field of deep learning and the like. The method comprises the following steps: acquiring an operator set of a neu...
Saved in:
Main Authors | , , , , , , , , |
---|---|
Format | Patent |
Language | Chinese English |
Published |
14.03.2023
|
Subjects | |
Online Access | Get full text |
Cover
Loading…
Summary: | The invention provides an operator fusion method and device, equipment and a storage medium, and relates to the technical field of artificial intelligence, in particular to the technical field of deep learning and the like. The method comprises the following steps: acquiring an operator set of a neural network model; performing topological sorting on each operator in the operator set to obtain a first sorting result; fusing the operators based on the first sorting result and a preset operator fusion rule to obtain a first fusion result, the first fusion result comprising at least one fused sub-graph; performing topological sorting on each sub-graph in the first fusion result to obtain a second sorting result; and fusing the sub-graphs based on the second sorting result and a preset sub-graph fusion rule to obtain a second fusion result. According to the operator fusion method provided by the invention, the operator fusion efficiency is improved, and the number of fused sub-graphs is reduced, so that the calcu |
---|---|
Bibliography: | Application Number: CN202211428372 |