Lines Matching refs:AffineMap
100 ArrayRef<AffineMap> indexingMaps, in replaceUnitDims()
124 SmallVector<AffineMap, 4> newIndexingMaps; in replaceUnitDims()
126 for (AffineMap operandMap : indexingMaps) { in replaceUnitDims()
142 newIndexingMaps, [](AffineMap map) -> Attribute { in replaceUnitDims()
174 SmallVector<AffineMap, 4> indexingMaps = genericOp.getIndexingMapsArray(); in matchAndRewrite()
181 AffineMap invertedMap = inversePermutation(concatAffineMaps(indexingMaps)); in matchAndRewrite()
224 AffineMap indexMap;
242 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in replaceUnitExtents()
277 reassociationMaps.push_back(AffineMapAttr::get(AffineMap::get( in replaceUnitExtents()
295 AffineMap::get(indexingMap.getNumDims(), in replaceUnitExtents()
369 SmallVector<AffineMap> newIndexingMaps; in matchAndRewrite()
391 AffineMap::get(origRank, /*symbolCount = */ 0, in matchAndRewrite()