SoftmaxFusionPass
融合模式
该融合规则将SoftmaxV2输入和输出添加TransposeD算子。
融合为
使用约束
输入是三维,属性axes的第一个值指定最后一根轴,且输入shape最后两维度大小是{8732, 21};
输入是五维,属性axes的第一个值指定最后一根轴,且输入shape最后四维度大小是{224, 224, 160, 4}。
当前只有这两个场景融合规则才生效。
父主题: 图融合规则说明
该融合规则将SoftmaxV2输入和输出添加TransposeD算子。
融合为
输入是三维,属性axes的第一个值指定最后一根轴,且输入shape最后两维度大小是{8732, 21};
输入是五维,属性axes的第一个值指定最后一根轴,且输入shape最后四维度大小是{224, 224, 160, 4}。
当前只有这两个场景融合规则才生效。