昇腾社区首页
中文
注册

MatMulBiasAddFusionPass

融合模式

将MatMul/MatMulv2/MatMulV3/BatchMatmul/BatchMatMulV2算子和biasadd/add算子融合为MatMul/MatMulv2/MatMulV3/BatchMatmul/BatchMatMulV2算子。

融合为

使用约束

  • add算子的两个输入必须得有一个的维度为1。
  • bias的维度与MatMul或MatMulV2或MatMulV3输出的n维保持一致。
  • MatMul/MatMulV2/MatMulV3的输出维度为2。