下载
中文
注册

Matmulv2FusionPass

融合模式

对于输入数量为3的MatMulV2算子,该融合规则在需要转置的输入TensorB之后插入TransposeD算子。

融合成

使用约束

  • MatMulV2的OpDesc必须有transpose_b这个布尔值,且该布尔值为True。
  • TensorB的维度数量应该为2,且数据类型为int8。