SparseSoftMaxFusionPass

Description

Fuses the SparseSoftmaxCrossEntropyWithLogits operator that fits the graph fusion pattern into the OneHotD and SoftmaxCrossEntropyWithLogits operators.

Restrictions

• The dynamic shape scenario is not supported.
• input0 supports only the float16 and float32 data types.
• input1 supports only the int32 data type.

Availability

Atlas A2 training product/Atlas A2 inference product