TbeAippConvReluMaxpoolingFusion
Description
Fuses the Aipp (optional)+Conv2D+Dequant (optional)+ElemWise (optional)+MaxPool (MaxPool/Pooling/MaxPoolv3)+AscendQuant (optional) operators in the following pattern subgraph into one operator.
Restrictions
Fusion can be performed only when the following conditions are met:
- Conv2D: Small channel is enabled. The kernel size must be 3 x 3, 5 x 5, or 7 x 7, strides must be [1, 1] or [2, 2], and cout must be less than or equal to 64.
- MaxPool: strides = [2, 2], ksize = [2, 2]/[3, 3].
- When MaxPool ksize is [2, 2] and the Conv2D input width is greater than 1000, the fusion pattern is disabled.
- When MaxPool ksize is [3, 3] and the Conv2D input width is greater than 800, the fusion pattern is disabled.
- Pooling: strides = [2, 2], window= [2, 2]/[3, 3].
- When Pooling window is [2, 2] and the Conv2D input width is greater than 1000, the fusion pattern is disabled.
- When Pooling window is [3, 3] and the Conv2D input width is greater than 800, the fusion pattern is disabled.
Availability
Parent topic: SpaceToBatchConv2dBatchToSpacePass