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));