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