昇腾社区首页
中文
注册

SoftmaxFusionPass

融合模式

该融合规则将SoftmaxV2输入和输出添加TransposeD算子。

融合为

使用约束

输入是三维,属性axes的第一个值指定最后一根轴,且输入shape最后两维度大小是{8732, 21};

输入是五维,属性axes的第一个值指定最后一根轴,且输入shape最后四维度大小是{224, 224, 160, 4}。

当前只有这两个场景融合规则才生效。