Searched refs:dimExprs (Results 1 – 6 of 6) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/ |
| H A D | NvGpuSupport.cpp | 197 auto makeMap = [&](ArrayRef<AffineExpr> dimExprs) -> AffineMap { in getLaneIdAndValueIdToOperandCoord() argument
198 return AffineMap::get(2, 0, dimExprs, builder.getContext()); in getLaneIdAndValueIdToOperandCoord()
248 auto makeMap = [&](ArrayRef<AffineExpr> dimExprs) -> AffineMap { in getLaneIdToLdMatrixMatrixCoord() argument
249 return AffineMap::get(1, 0, dimExprs, builder.getContext()); in getLaneIdToLdMatrixMatrixCoord()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransferOpTransforms.cpp | 452 SmallVector<AffineExpr, 1> dimExprs{ in matchAndRewrite() local
455 AffineMap::get(collapsedRank, 0, dimExprs, rewriter.getContext()); in matchAndRewrite()
510 SmallVector<AffineExpr, 1> dimExprs{ in matchAndRewrite() local
513 AffineMap::get(collapsedRank, 0, dimExprs, rewriter.getContext()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | Builders.cpp | 309 SmallVector<AffineExpr, 4> dimExprs; in getMultiDimIdentityMap() local
310 dimExprs.reserve(rank); in getMultiDimIdentityMap()
312 dimExprs.push_back(getAffineDimExpr(i)); in getMultiDimIdentityMap()
313 return AffineMap::get(/*dimCount=*/rank, /*symbolCount=*/0, dimExprs, in getMultiDimIdentityMap()
|
| H A D | AffineMap.cpp | 246 SmallVector<AffineExpr, 4> dimExprs; in getMultiDimIdentityMap() local
247 dimExprs.reserve(numDims); in getMultiDimIdentityMap()
249 dimExprs.push_back(mlir::getAffineDimExpr(i, context)); in getMultiDimIdentityMap()
250 return get(/*dimCount=*/numDims, /*symbolCount=*/0, dimExprs, context); in getMultiDimIdentityMap()
|
| /llvm-project-15.0.7/mlir/lib/Bindings/Python/ |
| H A D | IRAffine.cpp | 903 [](PyIntegerSet &self, py::list dimExprs, py::list symbolExprs, in populateIRAffine()
905 if (static_cast<intptr_t>(dimExprs.size()) != in populateIRAffine()
918 dimExprs, dimAffineExprs, in populateIRAffine()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalg.cpp | 1806 SmallVector<AffineExpr, 4> dimExprs; in matchAndRewrite() local
1807 dimExprs.reserve(rank); in matchAndRewrite()
1809 dimExprs.push_back(rewriter.getAffineDimExpr(i * 2 + 1)); in matchAndRewrite()
1812 AffineMap::get(/*dimCount=*/rank * 2, /*symbolCount=*/0, dimExprs, in matchAndRewrite()
|