Home
last modified time | relevance | path

Searched refs:nPar (Results 1 – 3 of 3) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DLoops.cpp187 auto nPar = copyOp.getNumParallelLoops(); in emitScalarImplementation() local
188 assert(nPar == allIvs.size()); in emitScalarImplementation()
190 permuteIvs(allIvs.take_front(nPar), copyOp.inputPermutation()); in emitScalarImplementation()
192 permuteIvs(allIvs.take_front(nPar), copyOp.outputPermutation()); in emitScalarImplementation()
199 nPar > 0 ? O(oivs) = I(iivs) : in emitScalarImplementation()
208 auto nPar = fillOp.getNumParallelLoops(); in emitScalarImplementation() local
209 assert(nPar == allIvs.size()); in emitScalarImplementation()
210 auto ivs = SmallVector<Value, 4>(allIvs.begin(), allIvs.begin() + nPar); in emitScalarImplementation()
214 nPar > 0 ? O(ivs) = fillOp.value() : O() = fillOp.value(); in emitScalarImplementation()
DFusion.cpp213 unsigned nPar = producer.getNumParallelLoops(); in fuse() local
216 SmallVector<Range, 8> loopRanges(nPar + nRed + nWin); in fuse()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
DLinalgStructuredOps.td143 unsigned nPar = getInputShapedType(0).getRank();
145 SmallVector<StringRef, 8>(nPar, getParallelIteratorTypeName()));
184 unsigned nPar = getOutputShapedType(0).getRank();
186 SmallVector<StringRef, 8>(nPar, getParallelIteratorTypeName()));
332 unsigned nPar = getOutputShapedType(0).getRank();
339 nPar - getNumBatchDimensions() - getNumInputFeatureDimensions();
340 SmallVector<StringRef, 8> iters(nPar, getParallelIteratorTypeName());
341 iters.reserve(nPar + nRed + nWin);
425 unsigned nPar = getOutputShapedType(0).getRank();
427 unsigned nWin = nPar;
[all …]