Lines Matching defs:collapsingInfo

1454                             const CollapsingInfo &collapsingInfo) {
1457        collapsingInfo.getCollapsedOpToOrigOpMapping()) {
1472                           const CollapsingInfo &collapsingInfo) {
1478       collapsingInfo.getOrigOpToCollapsedOpMapping();
1489   return AffineMap::get(collapsingInfo.getCollapsedOpIterationRank(), 0,
1497                         const CollapsingInfo &collapsingInfo) {
1501       collapsingInfo.getOrigOpToCollapsedOpMapping();
1503       collapsingInfo.getCollapsedOpToOrigOpMapping();
1525                                    const CollapsingInfo &collapsingInfo,
1529       getOperandReassociation(indexingMap, collapsingInfo);
1552                                      const CollapsingInfo &collapsingInfo,
1570        enumerate(collapsingInfo.getCollapsedOpToOrigOpMapping())) {
1590                                 const CollapsingInfo &collapsingInfo,
1598         return getCollapsedOpOperand(loc, op, opOperand, collapsingInfo,
1607         getCollapsedOpOperand(loc, op, &output, collapsingInfo, rewriter);
1619                         const CollapsingInfo &collapsingInfo) {
1627                                       const CollapsingInfo &collapsingInfo) {
1630   collapseOperandsAndResults(origOp, collapsingInfo, rewriter, inputOperands,
1642                                         const CollapsingInfo &collapsingInfo) {
1645   collapseOperandsAndResults(origOp, collapsingInfo, rewriter, inputOperands,
1649         return getCollapsedOpIndexingMap(map, collapsingInfo);
1653       origOp.getIteratorTypesArray(), collapsingInfo));
1665 LinalgOp createCollapsedOp(LinalgOp op, const CollapsingInfo &collapsingInfo,
1668     return cloneToCollapsedOp(rewriter, genericOp, collapsingInfo);
1670     return cloneToCollapsedOp(rewriter, op, collapsingInfo);
1698   CollapsingInfo collapsingInfo;
1700           collapsingInfo.initialize(op.getNumLoops(), foldedIterationDims))) {
1722   LinalgOp collapsedOp = createCollapsedOp(op, collapsingInfo, rewriter);
1734                                     collapsingInfo, loopBound, rewriter);
1749           getOperandReassociation(indexingMap, collapsingInfo);