| /external/llvm-project/mlir/lib/Dialect/Linalg/Utils/ |
| D | Utils.cpp | 128 ArrayRef<Attribute> iteratorTypes, in doit() argument
158 ArrayRef<Attribute> iteratorTypes, in doit() argument
204 ArrayRef<Attribute> iteratorTypes, in generateParallelLoopNest() argument
210 assert(lbs.size() == iteratorTypes.size());
214 if (iteratorTypes.empty())
218 unsigned nLoops = iteratorTypes.size();
220 nLoops - iteratorTypes.drop_while(isParallelIteratorType).size();
229 steps.drop_front(), iteratorTypes.drop_front(), in __anon7ecdf6bc0202()
245 steps.drop_front(nOuterPar), iteratorTypes.drop_front(nOuterPar), in __anon7ecdf6bc0302()
276 iteratorTypes.drop_front(numProcessed), bodyBuilderFn, ivStorage, in __anon7ecdf6bc0402()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Utils/ |
| D | StructuredOpsUtils.h | 107 inline unsigned getNumIterators(StringRef name, ArrayAttr iteratorTypes) { in getNumIterators() argument
111 return llvm::count_if(iteratorTypes, [name](Attribute a) { in getNumIterators()
116 inline unsigned getNumIterators(ArrayAttr iteratorTypes) { in getNumIterators() argument
119 res += getNumIterators(n, iteratorTypes); in getNumIterators()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/EDSC/ |
| D | Builders.cpp | 23 ArrayRef<IteratorType> iteratorTypes) { in vector_contraction() argument
29 llvm::to_vector<8>(llvm::map_range(iteratorTypes, toString))}); in vector_contraction()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | ElementwiseToLinalg.cpp | 42 SmallVector<StringRef, 6> iteratorTypes(rank, in matchAndRewrite() local
50 /*iteratorTypes=*/iteratorTypes, in matchAndRewrite()
|
| D | Tiling.cpp | 344 SmallVector<Attribute, 4> iteratorTypes; in tileLinalgOpImpl() local
348 iteratorTypes.push_back(attr.value()); in tileLinalgOpImpl()
371 applyPermutationToVector(iteratorTypes, interchangeVector); in tileLinalgOpImpl()
379 loopRanges, /*iterArgInitValues*/ initTensors, iteratorTypes, in tileLinalgOpImpl()
|
| D | DropUnitDims.cpp | 195 ArrayAttr iteratorTypes = op.iterator_types(); in matchAndRewrite() local
199 iteratorTypes[expr.index()].dyn_cast<StringAttr>().getValue() == in matchAndRewrite()
216 for (auto attr : llvm::enumerate(iteratorTypes)) { in matchAndRewrite()
|
| D | Vectorization.cpp | 587 std::vector<StringRef> iteratorTypes(numDims, "reduction"); in matchAndRewrite() local
592 rewriter.getStrArrayAttr(iteratorTypes)); in matchAndRewrite()
|
| D | Fusion.cpp | 532 static AffineMap pruneReductionDimsFromMap(ArrayRef<Attribute> iteratorTypes, in pruneReductionDimsFromMap() argument
535 for (auto attr : llvm::enumerate(iteratorTypes)) { in pruneReductionDimsFromMap()
|
| D | FusionOnTensors.cpp | 581 SmallVector<StringRef, 4> iteratorTypes(remapping.back(), in fuseWithReshapeByExpansion() local
588 /*initTensors=*/ValueRange{}, expandedOpIndexingMaps, iteratorTypes); in fuseWithReshapeByExpansion()
|
| D | Sparsification.cpp | 674 auto iteratorTypes = op.iterator_types().getValue(); in genFor() local
676 bool isParallel = linalg::isParallelIteratorType(iteratorTypes[idx]); in genFor()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 156 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build() argument
161 builder.getStrArrayAttr(iteratorTypes), in build()
182 ArrayRef<StringRef> iteratorTypes, StringRef doc, StringRef libraryCall, in build() argument
185 indexingMaps, iteratorTypes, doc, libraryCall, bodyBuild); in build()
191 ArrayRef<StringRef> iteratorTypes, in build() argument
193 build(builder, result, inputs, outputBuffers, indexingMaps, iteratorTypes, in build()
201 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build() argument
204 indexingMaps, iteratorTypes, in build()
211 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build() argument
217 builder.getStrArrayAttr(iteratorTypes), in build()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Vector/EDSC/ |
| D | Builders.h | 38 ArrayRef<IteratorType> iteratorTypes);
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 60 static SmallVector<Attribute, 4> adjustIter(ArrayAttr iteratorTypes, in adjustIter() argument
63 for (auto it : llvm::enumerate(iteratorTypes)) { in adjustIter()
1630 auto iteratorTypes = op.iterator_types().getValue(); in matchAndRewrite() local
1631 if (!isParallelIterator(iteratorTypes[0]) || in matchAndRewrite()
1632 !isParallelIterator(iteratorTypes[1]) || in matchAndRewrite()
1633 !isReductionIterator(iteratorTypes[2])) in matchAndRewrite()
1709 auto iteratorTypes = op.iterator_types().getValue(); in matchAndRewrite() local
1711 if (isParallelIterator(iteratorTypes[0]) && in matchAndRewrite()
1712 isParallelIterator(iteratorTypes[1]) && in matchAndRewrite()
1713 isReductionIterator(iteratorTypes[2])) { in matchAndRewrite()
[all …]
|
| D | VectorOps.cpp | 191 ArrayRef<StringRef> iteratorTypes) { in build() argument
198 builder.getStrArrayAttr(iteratorTypes)); in build()
204 ArrayAttr iteratorTypes) { in build() argument
208 result.addAttribute(getIteratorTypesAttrName(), iteratorTypes); in build()
454 getDimMap(ArrayRef<AffineMap> indexingMaps, ArrayAttr iteratorTypes, in getDimMap() argument
457 for (auto it : llvm::enumerate(iteratorTypes)) { in getDimMap()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOps.td | 643 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
647 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
652 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
655 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
783 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
788 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
794 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
798 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Utils/ |
| D | Utils.h | 220 ArrayRef<Attribute> iteratorTypes,
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/EDSC/ |
| D | Builders.h | 50 ArrayRef<IteratorType> iteratorTypes, ArrayRef<StructuredIndexed> inputs,
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/ |
| D | Builders.cpp | 25 ArrayRef<IteratorType> iteratorTypes, ArrayRef<StructuredIndexed> inputs, in makeGenericLinalgOp() argument
58 llvm::to_vector<8>(llvm::map_range(iteratorTypes, toString)); in makeGenericLinalgOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 307 llvm::SmallVector<mlir::IteratorType, 4> iteratorTypes( in EmitLinalgMatmul() local
309 iteratorTypes.push_back(mlir::IteratorType::Reduction); in EmitLinalgMatmul()
313 /*iteratorTypes=*/iteratorTypes, in EmitLinalgMatmul()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Vector/ |
| D | VectorOps.td | 153 "ArrayAttr":$indexingMaps, "ArrayAttr":$iteratorTypes)>,
156 "ArrayRef<StringRef>":$iteratorTypes)>
|