TbeConv3dElemwisePass

融合模式

该融合将满足如下Pattern关系的子图中Conv3D和Elemwise进行UB融合。

或者或者

使用约束

不支持动态shape场景。

支持3个pattern,对应上面3张示意图的场景:

  1. 融合两个Elemwise算子,第一个只支持Add,第二个只支持Relu。
  2. 融合一个Elemwise算子,并且Elemwise算子还有另外一路输入,Elemwise算子类型没有约束,并且支持AscendDequant和AscendRequant。
  3. 融合一个Elemwise算子,Elemwise算子仅一路输入无其他输入,该场景只支持Relu。

支持的型号

Atlas 200/300/500 推理产品

Atlas 推理系列产品

Atlas 训练系列产品