Lines Matching defs:RowColIndexing

431 struct RowColIndexing : private std::pair<AffineExpr, AffineExpr> {
432   RowColIndexing(AffineExpr row, AffineExpr col)
451       std::function<SmallVector<RowColIndexing>(MLIRContext *)>;
489   static SmallVector<RowColIndexing> m16n8k4tf32Lhs(MLIRContext *ctx) {
493     return {RowColIndexing{groupID, threadIDInGroup},
494             RowColIndexing{groupID + 8, threadIDInGroup}};
502   static SmallVector<RowColIndexing> m16n8k4tf32Rhs(MLIRContext *ctx) {
506     return {RowColIndexing{threadIDInGroup, groupID}};
515   static SmallVector<RowColIndexing> m16n8k4tf32Res(MLIRContext *ctx) {
519     return {RowColIndexing{groupID, threadIDInGroup * 2 + 0},
520             RowColIndexing{groupID, threadIDInGroup * 2 + 1},
521             RowColIndexing{groupID + 8, threadIDInGroup * 2 + 0},
522             RowColIndexing{groupID + 8, threadIDInGroup * 2 + 1}};
537   static SmallVector<RowColIndexing> m16n8k16f16Lhs(MLIRContext *ctx) {
543       RowColIndexing{groupID, threadIDInGroup * 2 + 0},         // i == 0
544       RowColIndexing{groupID, threadIDInGroup * 2 + 1},         // i == 1
545       RowColIndexing{groupID + 8, threadIDInGroup * 2 + 0},     // i == 2
546       RowColIndexing{groupID + 8, threadIDInGroup * 2 + 1},     // i == 3
547       RowColIndexing{groupID, threadIDInGroup * 2 + 0 + 8},     // i == 4
548       RowColIndexing{groupID, threadIDInGroup * 2 + 1 + 8},     // i == 5
549       RowColIndexing{groupID + 8, threadIDInGroup * 2 + 0 + 8}, // i == 6
550       RowColIndexing{groupID + 8, threadIDInGroup * 2 + 1 + 8}  // i == 7
563   static SmallVector<RowColIndexing> m16n8k16f16Rhs(MLIRContext *ctx) {
569       RowColIndexing{threadIDInGroup * 2 + 0, groupID},        // i == 0
570       RowColIndexing{threadIDInGroup * 2 + 1, groupID},        // i == 1
571       RowColIndexing{threadIDInGroup * 2 + 0 + 8, groupID},    // i == 2
572       RowColIndexing{threadIDInGroup * 2 + 1 + 8, groupID}     // i == 3
585   static SmallVector<RowColIndexing> m16n8k16f16Res(MLIRContext *ctx) {
591       RowColIndexing{groupID, threadIDInGroup * 2 + 0},        // i == 0
592       RowColIndexing{groupID, threadIDInGroup * 2 + 1},        // i == 1
593       RowColIndexing{groupID + 8, threadIDInGroup * 2 + 0},    // i == 2
594       RowColIndexing{groupID + 8, threadIDInGroup * 2 + 1}     // i == 3
671   SmallVector<RowColIndexing> indexings = indexFn(b.getContext());