下载
EN
注册

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 训练系列产品