BatchMatmulV2ReduceFusionPass

融合模式

该融合规则将BatchMatmulV2+Reduce算子融合为BatchMatmulV2Reduce算子。

模式一：

融合成

模式二：

融合成

使用约束

• BatchMatmul后节点若为Cast32，输出数据类型为fp16，Cast输出数据类型为fp32。
• BatchMatmul输出节点数只能为1。