MatmulReduceSumUbFusion

融合模式

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

使用约束

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

支持的型号

Atlas 200/300/500 推理产品

Atlas 200/500 A2推理产品

Atlas 推理系列产品

Atlas 训练系列产品