Searched defs:tidLvls (Results 1 – 4 of 4) sorted by relevance
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 817 shouldTryParallize(CodegenEnv & env,LoopId curr,ArrayRef<TensorLevel> tidLvls) shouldTryParallize() argument
834 genCoIteration(CodegenEnv & env,OpBuilder & builder,ArrayRef<TensorLevel> tidLvls,bool tryParallel,bool needsUniv) genCoIteration() argument
848 genLoop(CodegenEnv & env,OpBuilder & builder,LoopId curr,bool needsUniv,ArrayRef<TensorLevel> tidLvls) genLoop() argument
1093 SmallVector<TensorLevel> tidLvls; startLoopSeq() local
1154 translateBitsToTidLvlPairs(CodegenEnv & env,LatPointId li,LoopId curr,SmallVectorImpl<TensorLevel> & tidLvls,SmallVectorImpl<std::pair<TensorLevel,AffineExpr>> & affineTidLvls) translateBitsToTidLvlPairs() argument
1170 SmallVector<TensorLevel> tidLvls; startLoop() local
[all...] |
| H A D | SparseTensorRewriting.cpp | 1393 const SmallVector<TensorLevel, 1> tidLvls{ matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | LoopEmitter.h | 257 : tidLvls(tidLvls), loop(loop), userCodeBlock(userBlock), iv(iv) { member
|
| H A D | LoopEmitter.cpp | 396 ArrayRef<TensorLevel> tidLvls, SmallVectorImpl<SparseIterator *> &raIters, in categorizeIterators() argument
415 ArrayRef<TensorLevel> tidLvls) { in enterNewLoopSeq() argument
619 enterCoIterationOverTensorsAtLvls(OpBuilder & builder,Location loc,ArrayRef<TensorLevel> tidLvls,MutableArrayRef<Value> reduc,bool tryParallel,bool needsUniv) enterCoIterationOverTensorsAtLvls() argument
[all...] |