Searched refs:collapsedRank (Results 1 – 2 of 2) sorted by relevance
106 unsigned collapsedRank = collapsedType.getRank(); in verifyReshapeLikeTypes() local107 if (expandedRank < collapsedRank) in verifyReshapeLikeTypes()113 if (expandedRank == collapsedRank) in verifyReshapeLikeTypes()116 if (collapsedRank == 0) { in verifyReshapeLikeTypes()125 if (collapsedRank != op.getReassociation().size()) in verifyReshapeLikeTypes()127 << collapsedRank << ") to be the number of reassociation maps(" in verifyReshapeLikeTypes()
450 int64_t collapsedRank = collapsedSourceType.getRank(); in matchAndRewrite() local451 assert(collapsedRank == firstContiguousInnerDim + 1); in matchAndRewrite()455 AffineMap::get(collapsedRank, 0, dimExprs, rewriter.getContext()); in matchAndRewrite()508 int64_t collapsedRank = collapsedSourceType.getRank(); in matchAndRewrite() local509 assert(collapsedRank == firstContiguousInnerDim + 1); in matchAndRewrite()513 AffineMap::get(collapsedRank, 0, dimExprs, rewriter.getContext()); in matchAndRewrite()