Searched refs:srcRank (Results 1 – 6 of 6) sorted by relevance
44 auto srcRank = multiReductionOp.getSourceVectorType().getRank(); in matchAndRewrite() local55 for (int64_t i = 0; i < srcRank; ++i) in matchAndRewrite()82 SmallVector<bool> reductionMask(srcRank, false); in matchAndRewrite()85 reductionMask[srcRank - i - 1] = true; in matchAndRewrite()114 auto srcRank = multiReductionOp.getSourceVectorType().getRank(); in matchAndRewrite() local119 if (srcRank < 2) in matchAndRewrite()231 auto srcRank = multiReductionOp.getSourceVectorType().getRank(); in matchAndRewrite() local233 if (srcRank != 2) in matchAndRewrite()268 auto srcRank = multiReductionOp.getSourceVectorType().getRank(); in matchAndRewrite() local269 if (srcRank != 2) in matchAndRewrite()[all …]
124 int64_t srcRank = srcType.getRank(); in matchAndRewrite() local126 assert(dstRank >= srcRank); in matchAndRewrite()127 if (dstRank != srcRank) in matchAndRewrite()144 if (srcRank == 1) { in matchAndRewrite()
269 int64_t srcRank = srcType.getRank(); in matchAndRewrite() local273 if (srcRank <= 1 && dstRank == 1) { in matchAndRewrite()275 if (srcRank == 0) in matchAndRewrite()292 if (srcRank < dstRank) { in matchAndRewrite()307 assert(srcRank == dstRank); in matchAndRewrite()913 int64_t srcRank = sourceVectorType.getRank(); in matchAndRewrite() local915 if ((srcRank == 2 && resRank == 1) || (srcRank == 1 && resRank == 2)) in matchAndRewrite()924 for (int64_t r = 0; r < srcRank; r++) in matchAndRewrite()932 SmallVector<int64_t, 4> srcIdx(srcRank); in matchAndRewrite()938 incIdx(srcIdx, sourceVectorType, srcRank - 1); in matchAndRewrite()
244 int64_t srcRank = srcType.getRank(); in matchAndRewrite() local252 bool isResultCollapsed = srcRank > resultRank; in matchAndRewrite()303 int64_t srcRank = srcType.getRank(); in matchAndRewrite() local310 if (srcRank > resultRank) { in matchAndRewrite()
398 unsigned srcRank = srcTp.getRank(); in translateIndices() local401 bool isExpand = srcRank > dstRank; in translateIndices()442 assert((isExpand && i == dstRank) || (!isExpand && i == srcRank)); in translateIndices()467 unsigned srcRank = srcTp.getRank(); in genSparse2SparseReshape() local490 Value srcIdx = genAlloca(rewriter, loc, srcRank, rewriter.getIndexType()); in genSparse2SparseReshape()
1696 int64_t srcRank = srcVectorType.getRank(); in isBroadcastableTo() local1698 if (srcRank > dstRank) in isBroadcastableTo()1702 int64_t lead = dstRank - srcRank; in isBroadcastableTo()1703 for (int64_t r = 0; r < srcRank; ++r) { in isBroadcastableTo()2707 unsigned srcRank = srcVecType ? srcVecType.getRank() : 0; in matchAndRewrite() local2710 unsigned rankDiff = dstRank - srcRank; in matchAndRewrite()2715 for (unsigned i = 0; i < srcRank; i++) { in matchAndRewrite()4842 int64_t srcRank = srcType.getRank(); in verify() local4844 if (reductionDim >= srcRank) in verify()4850 if (initialValueRank != srcRank - 1) in verify()[all …]