Searched refs:getParallelIteratorTypeName (Results 1 – 11 of 11) sorted by relevance
| /external/llvm-project/mlir/include/mlir/Dialect/Utils/ |
| D | StructuredOpsUtils.h | 78 constexpr StringRef getParallelIteratorTypeName() { return "parallel"; } in getParallelIteratorTypeName() function
81 return strAttr && strAttr.getValue() == getParallelIteratorTypeName(); in isParallelIterator()
100 static constexpr StringRef names[3] = {getParallelIteratorTypeName(), in getAllIteratorTypeNames()
129 return getParallelIteratorTypeName(); in toString()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | ElementwiseToLinalg.cpp | 43 getParallelIteratorTypeName()); in matchAndRewrite()
|
| D | FusionOnTensors.cpp | 429 getParallelIteratorTypeName(); in isFusableWithReshapeByDimExpansion()
582 getParallelIteratorTypeName()); in fuseWithReshapeByExpansion()
|
| D | DropUnitDims.cpp | 200 getParallelIteratorTypeName()) in matchAndRewrite()
|
| D | Fusion.cpp | 664 getParallelIteratorTypeName(); in collectFusableLoops()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOps.td | 145 SmallVector<StringRef, 8>(nPar, getParallelIteratorTypeName()));
186 SmallVector<StringRef, 8>(nPar, getParallelIteratorTypeName()));
340 SmallVector<StringRef, 8> iters(nPar, getParallelIteratorTypeName());
428 SmallVector<StringRef, 8> iters(nPar, getParallelIteratorTypeName());
|
| D | LinalgStructuredOpsInterface.td | 35 return getNumIterators(getParallelIteratorTypeName(),
48 return getDimsOfType($_op, getParallelIteratorTypeName(), res);
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Utils/ |
| D | Utils.cpp | 62 return strAttr.getValue() == getParallelIteratorTypeName(); in isParallelIteratorType()
|
| /external/llvm-project/mlir/docs/Dialects/ |
| D | Linalg.md | 655 getParallelIteratorTypeName(),
656 getParallelIteratorTypeName(),
657 getParallelIteratorTypeName(),
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 58 getParallelIteratorTypeName()); in GetParallelAndReductionIterators()
951 : getParallelIteratorTypeName()); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 519 getParallelIteratorTypeName(), getContext()); in getBatchDimMap()
|