Lines Matching refs:AffineMap

154 static AffineMap getRegisterIndexToTileOffsetMap(int64_t lineSize,  in getRegisterIndexToTileOffsetMap()
165   return AffineMap::get(  in getRegisterIndexToTileOffsetMap()
172 FailureOr<AffineMap>
192   AffineMap registerIndexToTileCoord = getRegisterIndexToTileOffsetMap(  in getLaneIdAndValueIdToOperandCoord()
197   auto makeMap = [&](ArrayRef<AffineExpr> dimExprs) -> AffineMap {  in getLaneIdAndValueIdToOperandCoord()
198     return AffineMap::get(2, 0, dimExprs, builder.getContext());  in getLaneIdAndValueIdToOperandCoord()
237 FailureOr<AffineMap>
248   auto makeMap = [&](ArrayRef<AffineExpr> dimExprs) -> AffineMap {  in getLaneIdToLdMatrixMatrixCoord()
249     return AffineMap::get(1, 0, dimExprs, builder.getContext());  in getLaneIdToLdMatrixMatrixCoord()
286   auto infer = [](MapList m) { return AffineMap::inferFromExprList(m); };  in matchAndRewrite()
293   SmallVector<AffineMap, 4> maps = op.getIndexingMapsArray();  in matchAndRewrite()