GroupedMatmulTransFusionPass
融合模式
融合模式一:将Transpose或TransposeD从图中删除,并将weight转置信息打在算子属性上。如下图所示。
该融合模式支持的产品如下。
融合模式二:将weight前的Reshape+Transpose/TransposeD+Reshape从图中删除,并将weight转置信息打在算子属性上。如下图所示。
该融合模式支持的产品如下。
使用约束
- 只支持输入x为单tensor、weight为单tensor、y为单tensor的场景(单tensor表示tensorList输入中只有一个tensor)。
- Transpose/TransposeD支持第二轴和第三轴转置。
父主题: 图融合规则说明