昇腾社区首页
中文
注册

SplitConvConcatFusionPass

融合模式

对如下几种场景进行融合:

场景一

场景二

场景三

使用约束

  • 该融合结构会优先匹配ASplitConv2dConcatPass,不满足ASplitConv2dConcatPass约束条件才会匹配此融合pass。
  • ConcatD/ConcatV2D至少有两个不同输入节点。
  • ConcatD/ConcatV2D和SplitD/SplitVD的dim轴必须是C轴,C轴跟dtype对齐。
    • 原始dtype为fp16和float时,dim C需要为16的倍数。
    • 原始dtype为int8,dim C需要为32的倍数。
    • 原始dtype为int4,dim C需要为64的倍数。
  • ConcatD/ConcatV2D所有输入是4维,并且输入format为NCHW或者NHWC。
  • SplitD/SplitVD至少有两个不同输出节点。
  • ConcatD/ConcatV2D输入路数和SplitD/SplitVD输出路数一致。
  • SplitD/SplitVD连接的AscendQuant算子的scale、offset、sqrt_mode属性需要一致。

支持的型号

Atlas 200/300/500 推理产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品