Lines Matching refs:foldedDims
532 AffineMap foldedDims = reassociationMaps[resultExpr.index()]; in compute() local
533 numExpandedDims[pos] = foldedDims.getNumResults(); in compute()
535 expandedShape.slice(foldedDims.getDimPosition(0), numExpandedDims[pos]); in compute()
1143 for (const auto &foldedDims : in initialize() local
1145 for (const auto &dim : enumerate(foldedDims.value())) in initialize()
1147 std::make_pair<int64_t, unsigned>(foldedDims.index(), dim.index()); in initialize()
1311 for (auto &foldedDims : in generateCollapsedIndexingRegion() local
1313 ReassociationIndicesRef foldedDimsRef(foldedDims.value()); in generateCollapsedIndexingRegion()
1315 rewriter.create<linalg::IndexOp>(loc, foldedDims.index()); in generateCollapsedIndexingRegion()
1322 indexReplacementVals[foldedDims.value().front()] = newIndexVal; in generateCollapsedIndexingRegion()
1337 llvm::all_of(foldedIterationDims, [](ReassociationIndicesRef foldedDims) { in collapseGenericOpIterationDims() argument
1338 return foldedDims.size() <= 1; in collapseGenericOpIterationDims()