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.
  • For the biasadd node, both of its inputs must have static shapes. Otherwise, the fusion will not be applied.
  • For the biasadd node, the second input must be 1-dimensional. Otherwise, the fusion will not be applied.
  • No fusion is performed in the training scenario.

Availability

Atlas inference product

Atlas 200I/500 A2 inference product

Atlas training product

Atlas A2 training product/Atlas A2 inference product

Atlas A3 training product/Atlas A3 inference product