SpaceToBatchConv2dBatchToSpacePass

Description

Fuses the SpaceToBatchND, Conv2D, and BatchToSpaceND operators that are sequentially connected into one Conv2D operator.

Restrictions

• SpaceToBatchND/Conv2D/BatchToSpaceND can only have a single output.
• The weight dimension of BatchToSpaceND must be 2, including block_shape and crops.
• The crops value of BatchToSpaceND must be 0.
• The strides of Conv2D must be [1, 1, 1, 1].
• The pads of Conv2D must be [0, 0, 0, 0].
• The block_shape of SpaceToBatchND must be the same as that of BatchToSpaceND.
• Each dimension: dilations(Conv2D) x block_shape(BatchToSpaceND) ≤ 255.
• Each dimension: 0 ≤ padding(SpaceToBatchND) ≤ 255.

Availability

Atlas inference product

Atlas 200I/500 A2 inference product

Atlas training product