下载
EN
注册

TbeAippCommonFusionPass

融合模式

该融合将满足如下Pattern关系的子图中Aipp + Conv2D + ElemWise(可选) + AscendQuant(可选) + StridedWrite(可选)对应节点进行UB融合。

ElemWise可以有多个串连,最多支持5个。

使用约束

  • Conv2D算子, strides = [1, 1], pad = [0, 0, 0, 0], kernel 1*1 的场景不开启融合。
  • Aipp开启resize,不开启融合。
  • Aipp mode为dynamic,不开启融合。
  • Aipp的input format设置为["RGB16", "RGB20", "RGB24","RGB8_IR", "RGB16_IR","RGB24_IR"]中的一种时,不开启融合。
  • Aipp开启padding时,不开启融合。
  • 若Conv2D算子使能DMA,不支持融合。
  • 若给出最小Tiling,L1仍无法容纳Aipp处理结果,则放弃融合。
  • 若卷积的kernal H小于或者等于上下任意一方向的Conv2d pad与AIPP pad之和,则放弃融合。

支持的型号

Atlas 200/300/500 推理产品

Atlas 推理系列产品

Atlas 200/500 A2推理产品

Atlas 训练系列产品