Searched refs:reducedRank (Results 1 – 2 of 2) sorted by relevance
279 int reducedRank = getReducedRank(sourceType.getShape()); in matchAndRewrite() local280 if (reducedRank == sourceType.getRank()) in matchAndRewrite()282 if (reducedRank != vectorType.getRank()) in matchAndRewrite()291 SmallVector<Value> zeros(reducedRank, c0); in matchAndRewrite()292 auto identityMap = rewriter.getMultiDimIdentityMap(reducedRank); in matchAndRewrite()322 int reducedRank = getReducedRank(sourceType.getShape()); in matchAndRewrite() local323 if (reducedRank == sourceType.getRank()) in matchAndRewrite()325 if (reducedRank != vectorType.getRank()) in matchAndRewrite()334 SmallVector<Value> zeros(reducedRank, c0); in matchAndRewrite()335 auto identityMap = rewriter.getMultiDimIdentityMap(reducedRank); in matchAndRewrite()
440 size_t originalRank = originalShape.size(), reducedRank = reducedShape.size(); in computeRankReductionMask() local445 if (reducedIdx < reducedRank && in computeRankReductionMask()458 if (reducedIdx != reducedRank) in computeRankReductionMask()