Lines Matching refs:indexingMaps
15 bool mlir::isRowMajorMatmul(ArrayAttr indexingMaps) { in isRowMajorMatmul() argument
16 if (indexingMaps.size() != 3) in isRowMajorMatmul()
19 auto map0 = indexingMaps[0].cast<AffineMapAttr>().getValue(); in isRowMajorMatmul()
20 auto map1 = indexingMaps[1].cast<AffineMapAttr>().getValue(); in isRowMajorMatmul()
21 auto map2 = indexingMaps[2].cast<AffineMapAttr>().getValue(); in isRowMajorMatmul()
33 auto *context = indexingMaps.getContext(); in isRowMajorMatmul()
38 return indexingMaps == maps; in isRowMajorMatmul()
41 bool mlir::isColumnMajorMatmul(ArrayAttr indexingMaps) { in isColumnMajorMatmul() argument
42 if (indexingMaps.size() != 3) in isColumnMajorMatmul()
45 auto map0 = indexingMaps[0].cast<AffineMapAttr>().getValue(); in isColumnMajorMatmul()
46 auto map1 = indexingMaps[1].cast<AffineMapAttr>().getValue(); in isColumnMajorMatmul()
47 auto map2 = indexingMaps[2].cast<AffineMapAttr>().getValue(); in isColumnMajorMatmul()
59 auto *context = indexingMaps.getContext(); in isColumnMajorMatmul()
64 return indexingMaps == maps; in isColumnMajorMatmul()
67 bool mlir::isRowMajorBatchMatmul(ArrayAttr indexingMaps) { in isRowMajorBatchMatmul() argument
68 if (indexingMaps.size() != 3) in isRowMajorBatchMatmul()
71 auto map0 = indexingMaps[0].cast<AffineMapAttr>().getValue(); in isRowMajorBatchMatmul()
72 auto map1 = indexingMaps[1].cast<AffineMapAttr>().getValue(); in isRowMajorBatchMatmul()
73 auto map2 = indexingMaps[2].cast<AffineMapAttr>().getValue(); in isRowMajorBatchMatmul()
86 auto *context = indexingMaps.getContext(); in isRowMajorBatchMatmul()
91 return indexingMaps == maps; in isRowMajorBatchMatmul()