Lines Matching defs:foldedDims

645     AffineMap foldedDims = reassociationMaps[resultExpr.index()];
646     numExpandedDims[pos] = foldedDims.getNumResults();
648         expandedShape.slice(foldedDims.getDimPosition(0), numExpandedDims[pos]);
1396     for (const auto &foldedDims :
1398       for (const auto &dim : enumerate(foldedDims.value()))
1400             std::make_pair<int64_t, unsigned>(foldedDims.index(), dim.index());
1569   for (auto foldedDims :
1571     ReassociationIndicesRef foldedDimsRef(foldedDims.value());
1573         rewriter.create<linalg::IndexOp>(loc, foldedDims.index());
1580     indexReplacementVals[foldedDims.value().front()] = newIndexVal;
1680       llvm::all_of(foldedIterationDims, [](ReassociationIndicesRef foldedDims) {
1681         return foldedDims.size() <= 1;