Searched refs:numFoldedDims (Results 1 – 1 of 1) sorted by relevance
460 SmallVector<unsigned, 4> numFoldedDims(fusedIndexMap.getNumDims(), 1); in fuseWithReshapeByExpansion() local467 numFoldedDims[pos] = foldedDims.getNumResults(); in fuseWithReshapeByExpansion()469 expandedShape.slice(foldedDims.getDimPosition(0), numFoldedDims[pos]); in fuseWithReshapeByExpansion()501 SmallVector<unsigned, 4> remapping(numFoldedDims.size() + 1, 0); in fuseWithReshapeByExpansion()503 for (auto numFoldedDim : llvm::enumerate(numFoldedDims)) { in fuseWithReshapeByExpansion()538 reshapeDims, reshapeDims + numFoldedDims[origDim]); in fuseWithReshapeByExpansion()542 reshapeDims += numFoldedDims[origDim]; in fuseWithReshapeByExpansion()618 for (unsigned i : llvm::seq<unsigned>(0, numFoldedDims.size())) { in fuseWithReshapeByExpansion()633 llvm::seq<unsigned>(numFoldedDims.size(), argReplacements.size())) { in fuseWithReshapeByExpansion()