下载
中文
注册

BatchMatmulV2ReduceFusionPass

融合模式

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

模式一:

融合成

模式二:

融合成

使用约束

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