BatchMatmulFusionPass
融合模式
该融合规则将BatchMatMul/BatchMatMulV2/MatMul/MatMulV2算子融合修改为MatMul算子,提高计算性能。
融合为
使用约束
当BatchMatMul算子的两个输入shape维度都不大于2时,该融合生效。
父主题: 图融合规则说明
该融合规则将BatchMatMul/BatchMatMulV2/MatMul/MatMulV2算子融合修改为MatMul算子,提高计算性能。
融合为
当BatchMatMul算子的两个输入shape维度都不大于2时,该融合生效。