SoftmaxCrossEntropyWithLogitsPass

融合模式

该融合将符合图融合pattern的reshape + SoftmaxCrossEntropyWithLogits算子，融合后消掉reshape算子，仅保留SoftmaxCrossEntropyWithLogits算子。

使用约束

• 输入、输出参数的shape约束：

  input0、input2、input4的shape必须是4D。

• 输入、输出参数的format约束：

  input0、input2的format必须是NHWC，C<131072。

• 输入参数的顺序约束：
  • input0、input2必须是reshape的第一个输入。
  • SoftmaxCrossEntropyWithLogits的第一个输出必须是reshape的第一个输入。
• 性能说明：

  融合规则在C=1的场景时，性能有收益。

  如果有性能劣化，考虑关闭该融合规则。

• 输入参数的数据类型约束：

  input0、input2的数据类型保持一致。

  • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件：数据类型支持FLOAT32、BFLOAT16。

支持的型号

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