融合模式
该融合将满足如下Pattern关系的子图中BatchMatMul/GEMM和ElemWise进行UB融合。
模式一:

模式二:

模式三:

使用约束
- 不支持动态shape。
- 模式一Elemwise只支持FusedMulAdd,Add,Div,RealDiv,Relu,ReluGrad,Elemwise1只支持Add,Relu,FusedMulAdd。
- 模式二Elemwise只支持FusedMulAdd,Add,Div,RealDiv,Relu,ReluGrad。
- 模式三Elemwise只支持Mul,Elemwise1只支持Mul,Elemwise2只支持Sigmoid。
- BatchMatMul支持BatchMatMul,BatchMatMulV2。
支持的型号
Atlas 200/300/500 推理产品
Atlas 200I/500 A2 推理产品
Atlas 推理系列产品
Atlas 训练系列产品