SoftmaxSmallOpFusionPass

融合模式

该融合规则将ReduceMaxD、Sub等小算子组合识别并融合为SoftmaxV2算子。

融合为

使用约束

• 不支持动态shape场景。
• 数据类型限制：
  • Atlas 推理系列产品：数据类型支持FLOAT32、FLOAT16。
  • Atlas A2 训练系列产品/Atlas 800I A2 推理产品：数据类型支持FLOAT32、FLOAT16、BFLOAT16。

支持的型号

Atlas 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品