Searched refs:loopRanges (Results 1 – 6 of 6) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Utils/ |
| D | Utils.cpp | 127 ArrayRef<Range> loopRanges, ValueRange iterArgInitValues, in doit() argument
136 procInfo = distributionOptions->procInfo(builder, loc, loopRanges); in doit()
139 unpackRanges(loopRanges, lbs, ubs, steps); in doit()
157 ArrayRef<Range> loopRanges, ValueRange iterArgInitValues, in doit() argument
163 unpackRanges(loopRanges, lbs, ubs, steps); in doit()
315 ArrayRef<Range> loopRanges, ValueRange iterArgInitValues, in doit() argument
321 assert(iteratorTypes.size() >= loopRanges.size() && in doit()
323 iteratorTypes = iteratorTypes.take_front(loopRanges.size()); in doit()
332 unpackRanges(loopRanges, lbsStorage, ubsStorage, stepsStorage); in doit()
345 parallelLoopRanges.push_back(loopRanges[iteratorType.index()]); in doit()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Fusion.cpp | 61 ArrayRef<Range> loopRanges, in inferShapeComponents() argument
72 shapeRanges[idx++] = loopRanges[loopPos]; in inferShapeComponents()
91 ArrayRef<Range> loopRanges) { in cloneWithLoopRanges() argument
103 inferShapeComponents(map, loopRanges, offsets, sizes, strides); in cloneWithLoopRanges()
146 loopRanges[i].offset); in cloneWithLoopRanges()
216 SmallVector<Range, 8> loopRanges(nPar + nRed + nWin); in fuse() local
218 loopRanges[fusedLoops.first] = fusedLoops.second; in fuse()
223 for (unsigned i = 0, nLoops = loopRanges.size(); i < nLoops; ++i) { in fuse()
224 if (loopRanges[i].offset) in fuse()
226 << "existing LoopRange: " << loopRanges[i] << "\n"); in fuse()
[all …]
|
| D | Bufferize.cpp | 48 SmallVector<Range, 4> loopRanges; in allocateBuffersForResults() local
86 if (loopRanges.empty()) in allocateBuffersForResults()
87 loopRanges = linalgOp.createLoopRanges(b, loc); in allocateBuffersForResults()
94 Value size = maybeConvertToIndex(loc, loopRanges[loopIndex].size, b); in allocateBuffersForResults()
|
| D | Tiling.cpp | 340 SmallVector<Range, 4> loopRanges; in tileLinalgOpImpl() local
342 std::tie(loopRanges, loopIndexToRangeIndex) = makeTiledLoopRanges( in tileLinalgOpImpl()
370 applyPermutationToVector(loopRanges, interchangeVector); in tileLinalgOpImpl()
379 loopRanges, /*iterArgInitValues*/ initTensors, iteratorTypes, in tileLinalgOpImpl()
|
| D | Loops.cpp | 520 auto loopRanges = linalgOp.createLoopRanges(builder, op->getLoc()); in linalgOpToLoopsImpl() local
523 loopRanges, /*iterInitArgs*/ {}, linalgOp.iterator_types().getValue(), in linalgOpToLoopsImpl()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Utils/ |
| D | Utils.h | 219 doit(ArrayRef<Range> loopRanges, ValueRange iterArgInitValues,
|