SpaceToBatchConv2dBatchToSpacePass
融合模式
该融合将顺序连接的SpaceToBatch、Conv2D、BatchToSpace算子融合为Conv2D算子,如下图所示。
使用约束
- SpaceToBatch/Conv2D/BatchToSpace只能是单输出。
- BatchToSpace的权重维度必须为2,包括:block_shape和crops。
- BatchToSpace的crops必须都为0。
- Conv2D的strides必须等于[1, 1, 1, 1]。
- Conv2D的pads必须等于[0, 0, 0, 0]。
- SpaceToBatch和BatchToSpace的block_shape必须完全相等。
- 每个维度:dilations(Conv2D) * block_shape(BatchToSpace) <= 255。
- 每个维度:0 <= padding(SpaceToBatch) <= 255。
支持的型号
Atlas 200/300/500 推理产品
Atlas 推理系列产品
Atlas 200/500 A2推理产品
Atlas 训练系列产品
父主题: 图融合规则说明