ASplitConv2dConcatPass
融合模式
将Split/SplitV2 + Conv2D *N + Concat/ConcatV2融合成一个组卷积,简化图结构。
使用约束
- 每个Conv2D的输入个数必须相同,且大于等于2个。
- Conv2D输入的类型只支持:float、float16、int8和int32。
- 不支持Conv2D动态输入。
- 每个Conv2D的filter的shape和format要相同(只支持HWCN或NCHW)。
- 每个Conv2D只能单输出。
- Conv2D的filter和bias必须是类型{"Const", "Constant", "QuantBiasOptimization", "QuantWeightRollBack", "QuantBiasRollBack", "AscendWeightQuant"}中的一个。
- Split/SplitV2节点和Concat/ConcatV2节点的另一路输入必须是const类型,且Split/SplitV2节点的输出个数和Concat/ConcatV2的输入个数必须相同。
- Split/SplitV2节点的切分轴和Concat/ConcatV2节点的组合轴必须都为channel方向。
支持的型号
Atlas 200/300/500 推理产品
Atlas 推理系列产品
Atlas 200/500 A2推理产品
Atlas 训练系列产品
父主题: 图融合规则说明