Lines Matching defs:unitDims

230                        const llvm::SmallDenseSet<unsigned> &unitDims,
236     if (unitDims.count(indexOp.getDim()) != 0) {
241           unitDims, [&](unsigned dim) { return dim < indexOp.getDim(); });
411   llvm::SmallDenseSet<unsigned> unitDims;
416         unitDims.insert(expr.index());
428     if (unitDims.count(index)) {
528   replaceUnitDimIndexOps(replacementOp, unitDims, rewriter);
615     llvm::SmallDenseSet<unsigned> unitDims;
624         unitDims.insert(dim);
632     if (unitDims.empty()) {
639     while (dim < padRank && unitDims.contains(dim))
642       assert(!unitDims.contains(dim) && "expected non unit-extent");
646       while (dim < padRank && unitDims.contains(dim))
666         if (unitDims.contains(dim)) {