昇腾社区首页
中文
注册
开发者
下载

MulAddNPass

融合模式

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

使用约束

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

支持的型号

Atlas A2 训练系列产品

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