| /external/llvm-project/mlir/lib/Dialect/Linalg/Utils/ |
| D | Utils.cpp | 128 ArrayRef<Attribute> iteratorTypes, in doit()
158 ArrayRef<Attribute> iteratorTypes, in doit()
204 ArrayRef<Attribute> iteratorTypes, in generateParallelLoopNest()
316 ArrayRef<Attribute> iteratorTypes, in doit()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/EDSC/ |
| D | Builders.cpp | 23 ArrayRef<IteratorType> iteratorTypes) { in vector_contraction()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Utils/ |
| D | StructuredOpsUtils.h | 107 inline unsigned getNumIterators(StringRef name, ArrayAttr iteratorTypes) { in getNumIterators()
116 inline unsigned getNumIterators(ArrayAttr iteratorTypes) { in getNumIterators()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | ElementwiseToLinalg.cpp | 42 SmallVector<StringRef, 6> iteratorTypes(rank, in matchAndRewrite() local
|
| D | DropUnitDims.cpp | 195 ArrayAttr iteratorTypes = op.iterator_types(); in matchAndRewrite() local
|
| D | Tiling.cpp | 344 SmallVector<Attribute, 4> iteratorTypes; in tileLinalgOpImpl() local
|
| D | Vectorization.cpp | 587 std::vector<StringRef> iteratorTypes(numDims, "reduction"); in matchAndRewrite() local
|
| D | FusionOnTensors.cpp | 581 SmallVector<StringRef, 4> iteratorTypes(remapping.back(), in fuseWithReshapeByExpansion() local
|
| D | Fusion.cpp | 532 static AffineMap pruneReductionDimsFromMap(ArrayRef<Attribute> iteratorTypes, in pruneReductionDimsFromMap()
|
| D | Sparsification.cpp | 674 auto iteratorTypes = op.iterator_types().getValue(); in genFor() local
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 156 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
182 ArrayRef<StringRef> iteratorTypes, StringRef doc, StringRef libraryCall, in build()
191 ArrayRef<StringRef> iteratorTypes, in build()
201 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
211 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
241 ArrayRef<StringRef> iteratorTypes, StringRef doc, StringRef libraryCall, in build()
251 ArrayRef<StringRef> iteratorTypes, in build()
261 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/ |
| D | Builders.cpp | 25 ArrayRef<IteratorType> iteratorTypes, ArrayRef<StructuredIndexed> inputs, in makeGenericLinalgOp()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 60 static SmallVector<Attribute, 4> adjustIter(ArrayAttr iteratorTypes, in adjustIter()
1630 auto iteratorTypes = op.iterator_types().getValue(); in matchAndRewrite() local
1709 auto iteratorTypes = op.iterator_types().getValue(); in matchAndRewrite() local
1811 auto iteratorTypes = op.iterator_types().getValue(); in matchAndRewrite() local
|
| D | VectorOps.cpp | 191 ArrayRef<StringRef> iteratorTypes) { in build()
204 ArrayAttr iteratorTypes) { in build()
454 getDimMap(ArrayRef<AffineMap> indexingMaps, ArrayAttr iteratorTypes, in getDimMap()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 307 llvm::SmallVector<mlir::IteratorType, 4> iteratorTypes( in EmitLinalgMatmul() local
|