昇腾社区首页
中文
注册

TBEConvAddFusion

融合模式

该融合规则将Conv + Add 两个算子融合为一个 Conv 算子,提高计算性能:

或者

融合为

使用约束

  • Conv 只能单输出。
  • 如果是Conv3d的话 Add 只能单输出。
  • Conv 的 Bias 必须是常量。
  • Add 权重大小必须为1。
  • Add 的另一输入必须为常量。
  • Add 的另一输入大小不能小于或者等于0。