| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/ |
| H A D | StructuredOpsUtils.h | 109 inline unsigned getNumIterators(StringRef name, ArrayAttr iteratorTypes) { in getNumIterators()
118 inline unsigned getNumIterators(ArrayAttr iteratorTypes) { in getNumIterators()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ElementwiseToLinalg.cpp | 90 SmallVector<StringRef, 6> iteratorTypes(rank, in matchAndRewrite() local
|
| H A D | ElementwiseOpFusion.cpp | 775 SmallVector<StringRef> iteratorTypes(expansionInfo.getExpandedOpNumDims(), in fuseWithReshapeByExpansion() local
1027 auto iteratorTypes = genericOp.iterator_types().getValue(); in getCollapsableIterationSpaceDims() local
1200 getCollapsedOpIteratorTypes(ArrayRef<Attribute> iteratorTypes, in getCollapsedOpIteratorTypes()
1350 SmallVector<StringRef> iteratorTypes = getCollapsedOpIteratorTypes( in collapseGenericOpIterationDims() local
|
| H A D | Loops.cpp | 212 auto iteratorTypes = llvm::to_vector<4>(linalgOp.iterator_types().getValue()); in linalgOpToLoopsImpl() local
|
| H A D | DropUnitDims.cpp | 188 ArrayAttr iteratorTypes = genericOp.iterator_types(); in matchAndRewrite() local
|
| H A D | SplitReduction.cpp | 387 SmallVector<StringRef> iteratorTypes = in splitReductionByScaling() local
|
| H A D | Tiling.cpp | 409 SmallVector<Attribute, 4> iteratorTypes; in tileLinalgOpImpl() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 455 SmallVector<llvm::StringRef> iteratorTypes(transposeVector.size(), in makeTransposeOp() local
485 SmallVector<StringRef> iteratorTypes(memrefTypeTo.getRank(), in makeMemRefCopyOp() local
502 ArrayRef<Attribute> iteratorTypes, in doit()
562 ArrayRef<Attribute> iteratorTypes, in doit()
612 ValueRange steps, ArrayRef<Attribute> iteratorTypes, in generateParallelLoopNest()
727 ArrayRef<Attribute> iteratorTypes, in doit()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 671 ArrayAttr iteratorTypes, StringAttr doc, StringAttr libraryCall, in build()
699 ArrayRef<StringRef> iteratorTypes, StringRef doc, StringRef libraryCall, in build()
713 ArrayRef<StringRef> iteratorTypes, StringRef doc, StringRef libraryCall, in build()
723 ArrayRef<StringRef> iteratorTypes, in build()
734 ArrayRef<StringRef> iteratorTypes, in build()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/ |
| H A D | NvGpuSupport.cpp | 292 auto iteratorTypes = op.getIteratorTypes().getValue(); in matchAndRewrite() local
|
| H A D | VectorToGPU.cpp | 72 auto iteratorTypes = contract.getIteratorTypes().getValue(); in contractSupportsMMAMatrixType() local
293 auto iteratorTypes = op.getIteratorTypes().getValue(); in matchAndRewrite() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 39 SmallVector<StringRef> iteratorTypes(padOp.getResultType().getRank(), in getLoopIteratorTypes() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransforms.cpp | 55 static SmallVector<Attribute, 4> adjustIter(ArrayAttr iteratorTypes, in adjustIter()
181 ArrayAttr iteratorTypes) { in getReductionIndex()
989 SmallVector<StringRef> iteratorTypes; in matchAndRewrite() local
1364 auto iteratorTypes = op.getIteratorTypes().getValue(); in matchAndRewrite() local
1653 auto iteratorTypes = op.getIteratorTypes().getValue(); in matchAndRewrite() local
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalg.cpp | 828 SmallVector<StringRef, 4> iteratorTypes; in reduceMatchAndRewriteHelper() local
1985 SmallVector<StringRef, 4> iteratorTypes; in matchAndRewrite() local
|
| /llvm-project-15.0.7/mlir/tools/mlir-linalg-ods-gen/ |
| H A D | mlir-linalg-ods-yaml-gen.cpp | 124 SmallVector<LinalgIteratorTypeDef> iteratorTypes; member
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 336 auto iteratorTypes = op.iterator_types().getValue(); in isAdmissableTensorExp() local
1197 auto iteratorTypes = op.iterator_types().getValue(); in genFor() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 526 ArrayRef<StringRef> iteratorTypes) { in build()
539 ArrayAttr iteratorTypes) { in build()
547 ArrayAttr iteratorTypes, CombiningKind kind) { in build()
815 getDimMap(ArrayRef<AffineMap> indexingMaps, ArrayAttr iteratorTypes, in getDimMap()
|