Lines Matching refs:map2
21 auto map2 = indexingMaps[2].cast<AffineMapAttr>().getValue(); in isRowMajorMatmul() local
24 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isRowMajorMatmul()
25 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isRowMajorMatmul()
30 AffineExpr m = map2.getResult(0); in isRowMajorMatmul()
31 AffineExpr n = map2.getResult(1); in isRowMajorMatmul()
47 auto map2 = indexingMaps[2].cast<AffineMapAttr>().getValue(); in isColumnMajorMatmul() local
50 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isColumnMajorMatmul()
51 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isColumnMajorMatmul()
56 AffineExpr n = map2.getResult(0); in isColumnMajorMatmul()
57 AffineExpr m = map2.getResult(1); in isColumnMajorMatmul()
73 auto map2 = indexingMaps[2].cast<AffineMapAttr>().getValue(); in isRowMajorBatchMatmul() local
76 map2.getNumResults() != 3 || map0.getNumInputs() != 4 || in isRowMajorBatchMatmul()
77 map1.getNumInputs() != 4 || map2.getNumInputs() != 4) { in isRowMajorBatchMatmul()
82 AffineExpr b = map2.getResult(0); in isRowMajorBatchMatmul()
83 AffineExpr m = map2.getResult(1); in isRowMajorBatchMatmul()
84 AffineExpr n = map2.getResult(2); in isRowMajorBatchMatmul()