MatMulBiasAddFusionPass
融合模式
将matmul/matmulv2/batchmatmul/batchmatmulv2算子和biasadd/add算子融合为matmul/matmulv2batchmatmul/batchmatmulv2算子。
融合为
使用约束
1. add算子的两个输入必须得有一个的维度为1
2. bias的维度与matmul或matmulv2输出的n维保持一致
3. matmul/matmulv2的输出维度为2
父主题: 图融合规则说明
将matmul/matmulv2/batchmatmul/batchmatmulv2算子和biasadd/add算子融合为matmul/matmulv2batchmatmul/batchmatmulv2算子。
融合为
1. add算子的两个输入必须得有一个的维度为1
2. bias的维度与matmul或matmulv2输出的n维保持一致
3. matmul/matmulv2的输出维度为2