AReduceSumFusionPass

Description

Changes the ReduceSum operator that fits the graph fusion pattern to the Reshape operator in static scenarios.

Restrictions

  • The fusion pattern does not take effect in dynamic scenarios.
  • The axis is not an empty tensor, the axis size does not exceed the range of the input dimensions, and the input shape consists entirely of ones, for example, [1,1,1].
  • This fusion pattern cannot be disabled.

Availability

Atlas A2 training product/Atlas A2 inference product