昇腾社区首页
中文
注册

ConvBatchnormFusionPass

融合模式

该融合将Conv2d+batchnorm或者Conv3d+batchnorm融合为1个融合算子;当batchnorm的输入为2输入时,也可以将DepthwiseConv2d+batchnorm融合为1个融合算子。

融合为

使用约束

  • conv节点可以是Conv2D,也可以是 Conv3D。当batchnorm节点只有2个输入(mean,variance)时,conv节点也可以是DepthwiseConv2D。
  • batchnorm节点可以是batchnorm,也可以是BNInference。
  • filter、bias必须是const,否则不融合;如果filter是QuantWeightRollBack,不融合。
  • batchnorm节点的输入,可以是2个输入(mean,variance),也可以是上述4个输入,所有输入必须是const,否则不融合。
  • data输入为动态时,支持融合。