Searched defs:needsUniv (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 835 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
855 finalizeWhileOp(CodegenEnv & env,OpBuilder & builder,bool needsUniv) finalizeWhileOp() argument
1168 startLoop(CodegenEnv & env,OpBuilder & builder,LoopId curr,LatPointId li,bool needsUniv) startLoop() argument
1201 endLoop(CodegenEnv & env,RewriterBase & rewriter,Operation * loop,LatPointId li,bool needsUniv,bool isSingleCond) endLoop() argument
1250 bool needsUniv = startLoopSeq(env, rewriter, exp, curr, lts); genStmt() local
[all...] |
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | LoopEmitter.cpp | 526 MutableArrayRef<Value> reduc, bool needsUniv) { in emitWhileLoopOverTensorsAtLvls() argument
620 enterCoIterationOverTensorsAtLvls(OpBuilder & builder,Location loc,ArrayRef<TensorLevel> tidLvls,MutableArrayRef<Value> reduc,bool tryParallel,bool needsUniv) enterCoIterationOverTensorsAtLvls() argument
|