| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | BuiltinTypes.cpp | 554 layout = AffineMapAttr::get(AffineMap::getMultiDimIdentityMap( in get()
570 layout = AffineMapAttr::get(AffineMap::getMultiDimIdentityMap( in getChecked()
585 map = AffineMap::getMultiDimIdentityMap(shape.size(), in get()
605 map = AffineMap::getMultiDimIdentityMap(shape.size(), in getChecked()
623 map = AffineMap::getMultiDimIdentityMap(shape.size(), in get()
644 map = AffineMap::getMultiDimIdentityMap(shape.size(), in getChecked()
|
| H A D | AffineMap.cpp | 105 auto id = AffineMap::getMultiDimIdentityMap(dims, context); in getMinorIdentityMap()
244 AffineMap AffineMap::getMultiDimIdentityMap(unsigned numDims, in getMultiDimIdentityMap() function in AffineMap
|
| H A D | Builders.cpp | 308 AffineMap Builder::getMultiDimIdentityMap(unsigned rank) { in getMultiDimIdentityMap() function in Builder
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 210 AffineMap inputMap = b.getMultiDimIdentityMap(intermRank); in splitReduction()
252 AffineMap::getMultiDimIdentityMap(map.getNumDims(), op.getContext()); in scaleReductionDim()
265 AffineMap::getMultiDimIdentityMap(map.getNumDims(), op.getContext()); in insertParallelDim()
414 AffineMap map = b.getMultiDimIdentityMap(originalOutputType.getRank() + 1); in splitReductionByScaling()
|
| H A D | ElementwiseToLinalg.cpp | 89 rewriter.getMultiDimIdentityMap(rank)); in matchAndRewrite()
|
| H A D | Tiling.cpp | 190 loc, AffineMap::getMultiDimIdentityMap(vals.size(), loc.getContext()), in buildMax()
300 b, loc, AffineMap::getMultiDimIdentityMap(2, b.getContext()), in tileToForeachThreadOpImpl()
418 AffineMap::getMultiDimIdentityMap(tileSizes.size(), b.getContext()); in tileLinalgOpImpl()
591 auto map = AffineMap::getMultiDimIdentityMap(rank, b.getContext()); in tilePadOp()
|
| H A D | Split.cpp | 118 AffineMap::getMultiDimIdentityMap(/*numDims=*/2, rewriter.getContext()), in splitOp()
|
| H A D | DecomposeLinalgOps.cpp | 177 return rewriter.getMultiDimIdentityMap(domain.size()); in createPeeledGenericOp()
|
| H A D | Vectorization.cpp | 641 rewriter.getMultiDimIdentityMap(srcType.getRank())); in vectorizeCopy()
648 rewriter.getMultiDimIdentityMap(srcType.getRank())); in vectorizeCopy()
|
| H A D | Transforms.cpp | 718 rewriter.getMultiDimIdentityMap(inputShapedType.getRank()), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 268 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultTy.getRank())); in matchAndRewrite()
269 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultTy.getRank())); in matchAndRewrite()
425 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank)); in matchAndRewrite()
426 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank)); in matchAndRewrite()
597 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(outputTy.getRank())); in matchAndRewrite()
598 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(outputTy.getRank())); in matchAndRewrite()
820 auto affineMap = rewriter.getMultiDimIdentityMap(resultTy.getRank()); in matchAndRewrite()
|
| H A D | TosaToLinalg.cpp | 626 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(rank)); in elementwiseMatchAndRewriteHelper()
654 rewriter.getMultiDimIdentityMap(rank)); in elementwiseMatchAndRewriteHelper()
1136 rewriter.getMultiDimIdentityMap(resultTy.getRank())}; in matchAndRewrite()
1193 rewriter.getMultiDimIdentityMap(rank)}; in matchAndRewrite()
1241 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(rank)); in matchAndRewrite()
1370 rewriter.getMultiDimIdentityMap(resultTy.getRank())}; in matchAndRewrite()
1737 rewriter.getMultiDimIdentityMap(resultTy.getRank())}; in matchAndRewrite()
1816 readAffineMap, rewriter.getMultiDimIdentityMap(genericShape.size())}; in matchAndRewrite()
2073 rewriter.getMultiDimIdentityMap(resultTy.getRank())}; in matchAndRewrite()
2129 rewriter.getMultiDimIdentityMap(resultTy.getRank()), in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/CAPI/IR/ |
| H A D | AffineMap.cpp | 60 return wrap(AffineMap::getMultiDimIdentityMap(numDims, unwrap(ctx))); in mlirAffineMapMultiDimIdentityGet()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransferOpTransforms.cpp | 292 auto identityMap = rewriter.getMultiDimIdentityMap(reducedRank); in matchAndRewrite()
335 auto identityMap = rewriter.getMultiDimIdentityMap(reducedRank); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | Utils.cpp | 1146 if (oldMap != builder.getMultiDimIdentityMap(oldMap.getNumDims())) { in replaceAllMemRefUsesWith()
1173 indexRemap != builder.getMultiDimIdentityMap(indexRemap.getNumDims())) { in replaceAllMemRefUsesWith()
1205 auto newMap = builder.getMultiDimIdentityMap(newMemRefRank); in replaceAllMemRefUsesWith()
1403 auto composedMap = builder.getMultiDimIdentityMap(composedOpOperands.size()); in createAffineComputationSlice()
1807 .setLayout(AffineMapAttr::get(b.getMultiDimIdentityMap(newRank))); in normalizeMemRefType()
|
| H A D | LoopUtils.cpp | 2163 AffineMap fastBufferLayout = b.getMultiDimIdentityMap(rank); in generateCopy()
2216 auto memAffineMap = b.getMultiDimIdentityMap(memIndices.size()); in generateCopy()
2218 auto bufAffineMap = b.getMultiDimIdentityMap(bufIndices.size()); in generateCopy()
2244 auto tagAffineMap = b.getMultiDimIdentityMap(tagIndices.size()); in generateCopy()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 101 auto idMap = AffineMap::getMultiDimIdentityMap(2, b.getContext()); in bubbleUpPadSlice()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | AffineMap.h | 71 static AffineMap getMultiDimIdentityMap(unsigned numDims,
|
| H A D | Builders.h | 164 AffineMap getMultiDimIdentityMap(unsigned rank);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 222 boundMap = AffineMap::getMultiDimIdentityMap(1, value.getContext()); in getUpperBoundForIndex()
453 AffineMap::getMultiDimIdentityMap(transposeVector.size(), in makeTransposeOp()
484 AffineMap::getMultiDimIdentityMap(memrefTypeTo.getRank(), b.getContext()); in makeMemRefCopyOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | SuperVectorize.cpp | 1180 state.builder.getMultiDimIdentityMap(memRefType.getRank())) in vectorizeAffineLoad()
1225 state.builder.getMultiDimIdentityMap(memRefType.getRank())) in vectorizeAffineStore()
|
| /llvm-project-15.0.7/mlir/test/mlir-linalg-ods-gen/ |
| H A D | test-linalg-ods-yaml-gen.yaml | 244 # IMPL-NEXT: AffineMap tensorMap = AffineMap::getMultiDimIdentityMap(
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Vector/ |
| H A D | TestVectorTransforms.cpp | 471 auto map = AffineMap::getMultiDimIdentityMap(1, ctx); in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 2567 rank ? builder.getMultiDimIdentityMap(rank) : builder.getEmptyAffineMap(); in build()
2709 rank ? builder.getMultiDimIdentityMap(rank) : builder.getEmptyAffineMap(); in build()
3879 rank ? builder.getMultiDimIdentityMap(rank) : builder.getEmptyAffineMap(); in build()
3970 rank ? builder.getMultiDimIdentityMap(rank) : builder.getEmptyAffineMap(); in build()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1568 return AffineMap::getMultiDimIdentityMap(rank, context); in extractOrIdentityMap()
|