Lines Matching refs:unitDims
99 static ArrayAttr replaceUnitDims(DenseSet<unsigned> &unitDims, in replaceUnitDims() argument
112 if (unitDims.count(dim)) in replaceUnitDims()
132 numIterationDims - unitDims.size(), in replaceUnitDims()
149 const DenseSet<unsigned> &unitDims, in replaceUnitDimIndexOps() argument
155 if (unitDims.count(indexOp.dim()) != 0) { in replaceUnitDimIndexOps()
160 unitDims, [&](unsigned dim) { return dim < indexOp.dim(); }); in replaceUnitDimIndexOps()
186 DenseSet<unsigned> unitDims; in matchAndRewrite() local
192 unitDims.insert(expr.index()); in matchAndRewrite()
195 if (unitDims.empty()) in matchAndRewrite()
201 replaceUnitDims(unitDims, indexingMaps, context); in matchAndRewrite()
209 if (!unitDims.count(attr.index())) in matchAndRewrite()
216 replaceUnitDimIndexOps(genericOp, unitDims, rewriter); in matchAndRewrite()