Lines Matching refs:map2
26 AffineMap map2 = cast<AffineMapAttr>(indexingMaps[2]).getValue(); in isRowMajorMatmul() local
29 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isRowMajorMatmul()
30 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isRowMajorMatmul()
35 AffineExpr m = map2.getResult(0); in isRowMajorMatmul()
36 AffineExpr n = map2.getResult(1); in isRowMajorMatmul()
52 AffineMap map2 = cast<AffineMapAttr>(indexingMaps[2]).getValue(); in isColumnMajorMatmul() local
55 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isColumnMajorMatmul()
56 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isColumnMajorMatmul()
61 AffineExpr n = map2.getResult(0); in isColumnMajorMatmul()
62 AffineExpr m = map2.getResult(1); in isColumnMajorMatmul()
78 AffineMap map2 = cast<AffineMapAttr>(indexingMaps[2]).getValue(); in isRowMajorBatchMatmul() local
81 map2.getNumResults() != 3 || map0.getNumInputs() != 4 || in isRowMajorBatchMatmul()
82 map1.getNumInputs() != 4 || map2.getNumInputs() != 4) { in isRowMajorBatchMatmul()
87 AffineExpr b = map2.getResult(0); in isRowMajorBatchMatmul()
88 AffineExpr m = map2.getResult(1); in isRowMajorBatchMatmul()
89 AffineExpr n = map2.getResult(2); in isRowMajorBatchMatmul()
104 AffineMap map2 = cast<AffineMapAttr>(indexingMaps[2]).getValue(); in isVecmat() local
107 map2.getNumResults() != 1 || map0.getNumInputs() != 2 || in isVecmat()
108 map1.getNumInputs() != 2 || map2.getNumInputs() != 2) { in isVecmat()
114 AffineExpr n = map2.getResult(0); in isVecmat()
128 AffineMap map2 = cast<AffineMapAttr>(indexingMaps[2]).getValue(); in isBatchVecmat() local
131 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isBatchVecmat()
132 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isBatchVecmat()
139 AffineExpr n = map2.getResult(1); in isBatchVecmat()
153 AffineMap map2 = cast<AffineMapAttr>(indexingMaps[2]).getValue(); in isMatvec() local
156 map2.getNumResults() != 1 || map0.getNumInputs() != 2 || in isMatvec()
157 map1.getNumInputs() != 2 || map2.getNumInputs() != 2) { in isMatvec()
163 AffineExpr n = map2.getResult(0); in isMatvec()
177 AffineMap map2 = cast<AffineMapAttr>(indexingMaps[2]).getValue(); in isBatchMatvec() local
180 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isBatchMatvec()
181 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isBatchMatvec()
188 AffineExpr n = map2.getResult(1); in isBatchMatvec()