Lines Matching defs:rankDiff
1705 // `rankDiff` leading dimensions correspond to new broadcasted dims, drop the
1707 int64_t rankDiff = broadcastSrcRank - extractResultRank;
1709 std::next(extractPos.begin(), extractPos.size() - rankDiff));
2352 int64_t rankDiff = dstShape.size() - srcShape.size();
2353 int64_t dstDim = rankDiff;
2356 llvm::zip_equal(srcShape, dstShape.drop_front(rankDiff))) {
3219 unsigned rankDiff = destShape.size() - sourceShape.size();
3222 destVectorType.getScalableDims()[idx + rankDiff]) {
3228 auto destSize = destShape[idx + rankDiff];
3847 unsigned rankDiff = dstRank - srcRank;
3853 if (srcVecType.getDimSize(i) != dstVecType.getDimSize(i + rankDiff)) {
3867 getI64SubArray(op.getOffsets(), /* dropFront=*/rankDiff),
3868 getI64SubArray(op.getSizes(), /* dropFront=*/rankDiff),
3869 getI64SubArray(op.getStrides(), /* dropFront=*/rankDiff));