Searched refs:numExpandedDims (Results 1 – 1 of 1) sorted by relevance
528 SmallVector<unsigned> numExpandedDims(fusedIndexMap.getNumDims(), 1); in compute() local533 numExpandedDims[pos] = foldedDims.getNumResults(); in compute()535 expandedShape.slice(foldedDims.getDimPosition(0), numExpandedDims[pos]); in compute()546 for (const auto &numFoldedDim : llvm::enumerate(numExpandedDims)) { in compute()628 auto numExpandedDims = expansionInfo.getExpandedDims(dim).size(); in getReassociationForExpansion() local630 llvm::seq<int64_t>(numReshapeDims, numReshapeDims + numExpandedDims)); in getReassociationForExpansion()632 numReshapeDims += numExpandedDims; in getReassociationForExpansion()