Searched refs:foldedDims (Results 1 – 1 of 1) sorted by relevance
532 AffineMap foldedDims = reassociationMaps[resultExpr.index()]; in compute() local533 numExpandedDims[pos] = foldedDims.getNumResults(); in compute()535 expandedShape.slice(foldedDims.getDimPosition(0), numExpandedDims[pos]); in compute()1143 for (const auto &foldedDims : in initialize() local1145 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() local1313 ReassociationIndicesRef foldedDimsRef(foldedDims.value()); in generateCollapsedIndexingRegion()1315 rewriter.create<linalg::IndexOp>(loc, foldedDims.index()); in generateCollapsedIndexingRegion()1322 indexReplacementVals[foldedDims.value().front()] = newIndexVal; in generateCollapsedIndexingRegion()[all …]