MatMulReshapeBiasAddFusionPass

Description

Applies a Reshape operation to the output of the MatMul operator, followed by a BiasAdd/Add operation on the reshaped output.

Restrictions

• MatMul/MatMulV2 can have only two inputs.
• The input of BiasAdd/Add must be the outputs of Bias and Reshape.
• The Reshape node cannot split the last axis of the MatMul output.
• The shape of BiasAdd/Add is the last axis of the MatMul output.
• This fusion pattern takes effect only in the static graph mode.

Availability

Atlas A2 training product/Atlas A2 inference product

Atlas A3 training product/Atlas A3 inference product

Atlas 350 Accelerator Card