Lines Matching defs:RankReduceMatmul

1005 struct RankReduceMatmul : RankReduceContractionOps<FromOpTy, ToOpTy> {
1083   patterns.add<RankReduceMatmul<MatmulOp, VecmatOp>>(context);
1084   patterns.add<RankReduceMatmul<MatmulOp, MatvecOp>>(context);
1085   patterns.add<RankReduceMatmul<MatmulTransposeAOp, VecmatOp>>(context);
1086   patterns.add<RankReduceMatmul<MatmulTransposeBOp, MatvecOp>>(context);
1088   patterns.add<RankReduceMatmul<BatchMatmulOp, BatchVecmatOp>>(context);
1089   patterns.add<RankReduceMatmul<BatchMatmulOp, BatchMatvecOp>>(context);
1090   patterns.add<RankReduceMatmul<BatchMatmulTransposeAOp, BatchVecmatOp>>(
1092   patterns.add<RankReduceMatmul<BatchMatmulTransposeBOp, BatchMatvecOp>>(
1096   patterns.add<RankReduceMatmul<MatvecOp, DotOp>>(context);
1097   patterns.add<RankReduceMatmul<VecmatOp, DotOp>>(context);