Searched refs:foldedDim (Results 1 – 2 of 2) sorted by relevance
446 unsigned expandedDim = 0, foldedDim = 0; in matchAndRewrite() local450 foldedDim < foldedShape.size()) { in matchAndRewrite()451 int64_t dstSize = foldedShape[foldedDim]; in matchAndRewrite()454 reassociationExprs[foldedDim].push_back( in matchAndRewrite()459 reassociationExprs[foldedDim].push_back( in matchAndRewrite()463 if (foldedDim == foldedShape.size() - 1 || in matchAndRewrite()464 foldedShape[foldedDim + 1] != 1) { in matchAndRewrite()467 reassociationExprs[foldedDim].push_back( in matchAndRewrite()474 foldedDim++; in matchAndRewrite()
620 for (unsigned foldedDim = remapping[i] + 1; foldedDim != remapping[i + 1]; in fuseWithReshapeByExpansion() local621 foldedDim++) { in fuseWithReshapeByExpansion()623 expandedDimsShape[i][foldedDim - remapping[i]]; in fuseWithReshapeByExpansion()