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()132 numIterationDims - unitDims.size(), in replaceUnitDims()151 replaceBlockArgForUnitDimLoops(OpTy op, const DenseSet<unsigned> &unitDims, in replaceBlockArgForUnitDimLoops() argument158 IndexedGenericOp op, const DenseSet<unsigned> &unitDims, in replaceBlockArgForUnitDimLoops() argument164 for (unsigned unitDimLoop : unitDims) { in replaceBlockArgForUnitDimLoops()167 SmallVector<unsigned, 8> unitDimsToErase(unitDims.begin(), unitDims.end()); in replaceBlockArgForUnitDimLoops()194 DenseSet<unsigned> unitDims; in matchAndRewrite() local201 unitDims.insert(expr.index()); in matchAndRewrite()203 if (unitDims.empty()) in matchAndRewrite()[all …]