下载
中文
注册

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 训练系列产品