TbeConv3dElemwisePass
融合模式
该融合将满足如下pattern关系的子图中Conv3D和Elemwise进行UB融合。
或者
或者或者
使用约束
不支持动态shape场景。
支持4个pattern,对应上面4张示意图的场景:
- 融合两个Elemwise算子,第一个只支持Add,第二个只支持Relu。
- 融合一个Elemwise算子,并且Elemwise算子还有另外一路输入,Elemwise算子类型没有约束,并且支持AscendDequant和AscendRequant。
- 融合一个Elemwise算子,Elemwise算子仅一路输入无其他输入,该场景只支持Relu。.
- 融合三个Elemwise算子,第一个只支持AscendDequant,第二个只支持Add,第三个只支持Relu。该场景仅支持
Atlas 推理系列产品 。
支持的型号
父主题: UB融合规则说明