ZConfusionSoftmaxGradFusionPass

融合模式

该融合将符合图融合pattern的Mul + ReduceSumD + Sub这几个小算子融合成ConfusionSoftmaxGrad算子。

使用约束

• 动态shape场景，该融合规则不生效。
• 当Mul算子的两个输入shape不相同时，该融合规则不生效。
• 当Mul算子的input0输入与Sub算子的input0输入不是同一个节点时，该融合规则不生效。
• 当input0不是Sub的第一个输入参数时，该融合规则不生效。
• 当ReduceSumD的属性axis不是其输入shape的尾轴索引时，该融合规则不生效。
• 当ReduceSumD输入shape的尾轴大于30000时，该融合规则不生效。
• 数据类型不在融合后算子的支持范围内，该融合规则不生效。
• input0、input1的输入数据类型和数据格式保持一致。
• 在reduce轴为1且AReduceSumFusionPass融合规则开启时，当前融合规则不生效。

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件