StridedSliceConcatFusionPass
融合模式
该融合规则将StridedSliceD/StridedSliceD/ConcatD算子融合成Squeeze/Roll/Unsqueeze,提高计算性能。如下图所示:
使用约束
满足如下条件时,该融合规则不生效。
- concat_dim不为1。
- 两个StridedSliceD的输入不为同一个。
- 输入的shape不为4维。
- 输入shape的最后两维不是32位对齐。
- 属性strides的值不全为1。
支持的型号
Atlas 推理系列产品
Atlas 训练系列产品
父主题: 图融合规则说明