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 训练系列产品
父主题: 图融合规则说明