Lines Matching defs:loopIdx

229 // Update the given `shardingOption` according to `meshAxes` and `loopIdx`
234                                         unsigned loopIdx) {
236       (!shardingOption.shardingArray[loopIdx].empty() &&
237        shardingOption.shardingArray[loopIdx] != meshAxes)) {
239                       << loopIdx << "\n");
243     if (i == loopIdx)
256   if (shardingOption.shardingArray[loopIdx].empty())
257     shardingOption.shardingArray[loopIdx].append(meshAxes.begin(),
313       for (size_t loopIdx = 0; loopIdx < loopTypes.size(); ++loopIdx) {
314         if (isReductionLoop(loopTypes[loopIdx]))
315           visitedLoopIndices.insert(loopIdx);
347         unsigned loopIdx = *loopIndices->begin();
348         visitedLoopIndices.insert(loopIdx);
350                                       shardAttr.getMeshAttr(), axes, loopIdx)))
358         for (unsigned loopIdx : *loopIndices) {
359           if (visitedLoopIndices.contains(loopIdx)) {
418     unsigned loopIdx = dim.getPosition();
419     if (loopIdx < shardingOption.shardingArray.size())
420       splitAxes[it.index()].append(shardingOption.shardingArray[loopIdx]);
462     for (unsigned loopIdx : *loopIndices) {
463       if ((size_t)loopIdx < shardingOption.shardingArray.size() &&
464           !shardingOption.shardingArray[loopIdx].empty())
465         shardedLoopIndices.push_back(loopIdx);