Lines Matching refs:collapsingInfo
1201 const CollapsingInfo &collapsingInfo) { in getCollapsedOpIteratorTypes() argument
1204 collapsingInfo.getCollapsedOpToOrigOpMapping()) { in getCollapsedOpIteratorTypes()
1220 const CollapsingInfo &collapsingInfo) { in getCollapsedOpIndexingMap() argument
1226 collapsingInfo.getOrigOpToCollapsedOpMapping(); in getCollapsedOpIndexingMap()
1237 return AffineMap::get(collapsingInfo.getCollapsedOpIterationRank(), 0, in getCollapsedOpIndexingMap()
1245 const CollapsingInfo &collapsingInfo) { in getOperandReassociation() argument
1249 collapsingInfo.getOrigOpToCollapsedOpMapping(); in getOperandReassociation()
1251 collapsingInfo.getCollapsedOpToOrigOpMapping(); in getOperandReassociation()
1273 const CollapsingInfo &collapsingInfo, in getCollapsedOpOperand() argument
1277 getOperandReassociation(indexingMap, collapsingInfo); in getCollapsedOpOperand()
1294 const CollapsingInfo &collapsingInfo, in generateCollapsedIndexingRegion() argument
1312 enumerate(collapsingInfo.getCollapsedOpToOrigOpMapping())) { in generateCollapsedIndexingRegion()
1342 CollapsingInfo collapsingInfo; in collapseGenericOpIterationDims() local
1343 if (failed(collapsingInfo.initialize(genericOp.getNumLoops(), in collapseGenericOpIterationDims()
1351 genericOp.iterator_types().getValue(), collapsingInfo); in collapseGenericOpIterationDims()
1356 return getCollapsedOpIndexingMap(map, collapsingInfo); in collapseGenericOpIterationDims()
1364 return getCollapsedOpOperand(loc, genericOp, opOperand, collapsingInfo, in collapseGenericOpIterationDims()
1375 getCollapsedOpOperand(loc, genericOp, output, collapsingInfo, rewriter); in collapseGenericOpIterationDims()
1406 collapsingInfo, loopBound, rewriter); in collapseGenericOpIterationDims()
1422 getOperandReassociation(indexingMap, collapsingInfo); in collapseGenericOpIterationDims()