SoftmaxFusionPass

Description

Adds TransposeD operators to the input and output of SoftmaxV2.

Before: After:

Restrictions

When the input is 3D, the first value of the axes attribute specifies the last axis, and the last two dimensions of the input shape are {8732, 21}.

When the input is 5D, the first value of the axes attribute specifies the last axis, and the last four dimensions of the input shape are {224, 224, 160, 4}.

Currently, fusion patterns take effect only in the two scenarios.