Lines Matching defs:tidLvls
827 ArrayRef<TensorLevel> tidLvls) {
830 bool isSparse = llvm::any_of(tidLvls, [curr, &env](TensorLevel tidLvl) {
844 ArrayRef<TensorLevel> tidLvls,
850 builder, env.op().getLoc(), tidLvls, numCases, reduc, tryParallel,
861 ArrayRef<TensorLevel> tidLvls) {
862 bool tryParallel = shouldTryParallize(env, curr, tidLvls);
863 return genCoIteration(env, builder, tidLvls, numCases, tryParallel,
1127 SmallVector<TensorLevel> tidLvls;
1132 if (llvm::find(tidLvls, tl) != tidLvls.end())
1134 tidLvls.emplace_back(tl);
1137 env.emitter().enterNewLoopSeq(builder, env.op().getLoc(), tidLvls);
1188 SmallVectorImpl<TensorLevel> &tidLvls,
1195 tidLvls.emplace_back(tl);
1207 SmallVector<TensorLevel> tidLvls;
1213 translateBitsToTidLvlPairs(env, li, curr, tidLvls, affineTidLvls);
1216 Operation *loop = genLoop(env, builder, curr, numCases, needsUniv, tidLvls);
1226 llvm::concat<TensorLevel>(tidLvls, llvm::make_first_range(affineTidLvls));