Lines Matching refs:AffineExpr
159 AffineExpr logicalValueId) { in getRegisterIndexToTileOffsetMap()
164 AffineExpr registerIdx = logicalValueId.floorDiv(elementsPerRegister); in getRegisterIndexToTileOffsetMap()
187 AffineExpr laneId, logicalValueIdDim; in getLaneIdAndValueIdToOperandCoord()
197 auto makeMap = [&](ArrayRef<AffineExpr> dimExprs) -> AffineMap { in getLaneIdAndValueIdToOperandCoord()
246 AffineExpr d0 = getAffineDimExpr(0, builder.getContext()); in getLaneIdToLdMatrixMatrixCoord()
248 auto makeMap = [&](ArrayRef<AffineExpr> dimExprs) -> AffineMap { in getLaneIdToLdMatrixMatrixCoord()
254 AffineExpr row = d0 % (operandShape[0]); in getLaneIdToLdMatrixMatrixCoord()
255 AffineExpr col = d0.floorDiv(operandShape[0]) * (kElementsPer128b); in getLaneIdToLdMatrixMatrixCoord()
285 using MapList = ArrayRef<ArrayRef<AffineExpr>>; in matchAndRewrite()
287 AffineExpr m; in matchAndRewrite()
288 AffineExpr n; in matchAndRewrite()
289 AffineExpr k; in matchAndRewrite()