昇腾社区首页
中文
注册

MatMulReshapeBiasAddFusionPass

融合模式

该融合规则对MatMul算子的输出做了reshape处理,并对Reshape输出做BiasAdd/Add处理。

使用约束

  • MatMul/MatMulV2必须且仅支持两个输入。
  • BiasAdd/Add的输入必须是一个Bias和一个reshape的输出。
  • Reshape节点无法拆分MatMul节点输出的尾轴。

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

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