Lines Matching defs:smeTile

111                                            SMESubTile smeTile) {
113                                    {smeTile.row, smeTile.col});
125                      SMESubTile smeTile) {
134       builder, loc, createMask.getOperands(), {-smeTile.row, -smeTile.col});
136       loc, smeTile.type.clone(builder.getI1Type()), smeTileMaskDims);
236     for (auto [index, smeTile] : llvm::enumerate(
239       auto smeMask = extractSMEMask(rewriter, loc, mask, smeTile);
241           loc, sliceType, outerProductOp.getLhs(), smeTile.row);
243           loc, sliceType, outerProductOp.getRhs(), smeTile.col);
314     for (SMESubTile smeTile :
316       auto smeMask = extractSMEMask(rewriter, loc, mask, smeTile);
319           getSMESubTileIndices(rewriter, loc, readOp.getIndices(), smeTile),
363     for (auto [index, smeTile] : llvm::enumerate(decomposeToSMETiles(
365       auto smeMask = extractSMEMask(rewriter, loc, mask, smeTile);
368           getSMESubTileIndices(rewriter, loc, writeOp.getIndices(), smeTile),
469     for (auto [index, smeTile] : llvm::enumerate(
472       auto tileRow = createVscaleMultiple(smeTile.row);
473       auto tileCol = createVscaleMultiple(smeTile.col);
492               loc, sliceMaskType, sliceMask, smeTile.col);
859     for (auto [index, smeTile] : llvm::enumerate(
869             rewriter.getIndexAttr(d + smeTile.row));
872               loc, smeSliceType, vector, smeTile.col);
878       auto transposedRow = createVscaleMultiple(smeTile.col);
880           rewriter.create<arith::ConstantIndexOp>(loc, smeTile.row);