下载
EN
注册

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输入为动态时,支持融合。
  • 卷积节点weight输入为动态时,不融合。

支持的型号

Atlas 200/300/500 推理产品

Atlas 推理系列产品

Atlas 200/500 A2推理产品

Atlas 训练系列产品