Conv2DSqueezeBiasaddFusionPass

Description

Converts the conv2D+squeeze+biasadd operators into the conv2D+biasadd+squeeze operators.

Before:

After:

Restrictions

• The data node input to the biasadd node must be 1-dimensional. Otherwise, an error is reported.
• The input node to the biasadd node must be data. If it is a Variable node instead, fusion fails.
• The two inputs of the biasadd node must have static shapes. Otherwise, the two inputs are not fused.
• The second input dimension of the biasadd node must be 1. Otherwise, the fusion fails.
• No fusion is performed in the training scenario.

Availability

Atlas 200/300/500 Inference Product

Atlas Training Series Product