下载
EN
注册

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 训练系列产品