MatmulReduceSumUbFusion
融合模式
该融合将满足如下Pattern关系的子图中BatchMatMul和ReduceSum进行UB融合。

使用约束
BatchMatMul的输入都有且都不为1;输出为1维;reduce的输出为float32类型,keep_dim为false的情况下,进行UB融合。
父主题: UB融合规则说明
该融合将满足如下Pattern关系的子图中BatchMatMul和ReduceSum进行UB融合。

BatchMatMul的输入都有且都不为1;输出为1维;reduce的输出为float32类型,keep_dim为false的情况下,进行UB融合。