昇腾社区首页
中文
注册

工具实现的融合功能

当前该工具主要实现的融合功能如下:

  • Conv+BN融合:AMCT在量化前会对模型中的"Conv+BatchNormalization"结构做Conv+BN融合,融合后的"BatchNormalization"层会被删除。
  • BatchNormalization+Mul:AMCT在量化前会对模型中的"BatchNormalization+Mul"结构做"BN+Mul"融合,融合后的"Mul"层会被删除。
  • BatchNormalization+Add:AMCT在量化前会对模型中的"BatchNormalization+Add"结构做"BN+Add"融合,融合后的"Add"层会被删除。