ASplitConv2dConcatPass

Description

Fuses the Split/SplitV2+Conv2D*N+Concat/ConcatV2 operators into a group convolution to simplify the graph structure.

Restrictions

• The number of inputs of each Conv2D must be the same and greater than or equal to 2.
• The Conv2D input can only be of float, float16, int8, or int32 type.
• Conv2D does not support dynamic inputs.
• The shape and format of each Conv2D filter must be the same (only HWCN or NCHW is supported).
• Each Conv2D supports only one output.
• The filter and bias of Conv2D must be of any of the following types: {"Const", "Constant", "QuantBiasOptimization", "QuantWeightRollBack", "QuantBiasRollBack", "AscendWeightQuant"}.
• The Split/SplitV2 node and the other input of the Concat/ConcatV2 node must be of const type, and the number of outputs of the Split/SplitV2 node must be the same as the number of inputs of the Concat/ConcatV2 node.
• The split axis of the Split/SplitV2 node and the combination axis of the Concat/ConcatV2 node must be in the channel direction.

Availability

Atlas 200/300/500 Inference Product

Atlas Training Series Product