Searched refs:unitDims (Results 1 – 1 of 1) sorted by relevance
99 static ArrayAttr replaceUnitDims(DenseSet<unsigned> &unitDims, in replaceUnitDims() argument112 if (unitDims.count(dim)) in replaceUnitDims()149 const DenseSet<unsigned> &unitDims, in replaceUnitDimIndexOps() argument155 if (unitDims.count(indexOp.dim()) != 0) { in replaceUnitDimIndexOps()160 unitDims, [&](unsigned dim) { return dim < indexOp.dim(); }); in replaceUnitDimIndexOps()186 DenseSet<unsigned> unitDims; in matchAndRewrite() local192 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()[all …]