昇腾社区首页
中文
注册

muladdnfusionpass

融合模式

该融合规则在mul节点个数大于2个时,将多个mul节点和addn节点融合成一个MulAddn节点。

使用约束

  • 输入为动态shape,x1的输入为[B,M,1],x2为[B, 1,N]。
  • mul节点个数大于2个。
  • N最大值为2040。

支持的型号

Atlas A2 训练系列产品

Atlas A3 训练系列产品/Atlas A3 推理系列产品