TbeConv3dElemwisePass

Description

Performs UB fusion on Conv3D and Elemwise in the following pattern subgraphs.

or

or or

Restrictions

The dynamic shape scenario is not supported.

Four patterns are supported, corresponding to the scenarios shown in the preceding four figures.

1. Fuses two Elemwise operators. The first operator supports only Add, and the second operator supports only ReLU.
2. Fuses an Elemwise operator that has another input. The Elemwise operator type is not restricted. AscendDequant and AscendRequant are supported.
3. Fuses an Elemwise operator that has only one input. Only ReLU is supported. .
4. Fuses three Elemwise operators. The first operator supports only AscendDequant, the second supports only Add, and the third supports only ReLU. This scenario is supported only by Atlas inference product.

Availability

Atlas inference product

Atlas training product