AConv2dMulFusion
融合模式
该融合将Conv2d+mul或Conv3d+mul融合为1个融合算子Conv。
融合为
使用约束
- conv节点可以是Conv2D,也可以是Conv3D。
- data输入为动态时,支持融合。
- filter、bias和mul的另一路,三个输入均为const时,支持融合。
- 当conv节点是Conv3D时,仅支持data输入的format为NDHWC格式,此时mul算子的const mul输入为1维输入,且该维度等于Conv3D算子输出的C维度时,才会进行融合。
- 当conv节点是Conv3D时,仅支持data输入的format为NDHWC格式,此时mul算子的const mul输入也为NDHWC格式,且NDHW维度均为1,只有C维度等于Conv3D算子输出的C维度时,才会进行融合。
支持的型号
Atlas 200/300/500 推理产品
Atlas 推理系列产品
Atlas 200/500 A2推理产品
Atlas 训练系列产品
父主题: 图融合规则说明