MatmulReshapeFusionPass
融合模式
该融合规则在Matmul N轴等于1时,会对第二个输入增加Reshape节点。
使用约束
- 该规则默认关闭,可参考如何关闭/开启融合规则自行开启。
- 仅支持静态shape场景。
- 支持数据类型:BFLOAT16、FLOAT16、FLOAT32。
- x1的输入为[M, K],x2的输入未转置,N=1并在尾轴,输入为[K, 1]。
- 数据类型为BFLOAT16或FLOAT16时,K<27392时,该融合规则生效。
支持的型号
父主题: 图融合规则说明