Searched refs:rankDiff (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/lib/Conversion/ShapeToStandard/ |
| D | ConvertShapeConstraints.cpp | 55 Value rankDiff = in matchAndRewrite() local
65 loc, rankDiff, greaterRank, one, llvm::None, in matchAndRewrite()
69 Value ivShifted = b.create<SubIOp>(loc, indexTy, iv, rankDiff); in matchAndRewrite()
|
| D | ShapeToStandard.cpp | 113 Value rankDiff = in matchAndRewrite() local
120 loc, CmpIPredicate::ult, outputDimension, rankDiff); in matchAndRewrite()
142 b.create<SubIOp>(loc, indexTy, outputDimension, rankDiff); in matchAndRewrite()
255 Value rankDiff = in matchAndRewrite() local
263 loc, rankDiff, greaterRank, one, ValueRange{init}, in matchAndRewrite()
269 Value ivShifted = b.create<SubIOp>(loc, indexTy, iv, rankDiff); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/VectorToLLVM/ |
| D | ConvertVectorToLLVM.cpp | 993 int64_t rankDiff = dstType.getRank() - srcType.getRank(); in matchAndRewrite() local
994 assert(rankDiff >= 0); in matchAndRewrite()
995 if (rankDiff == 0) in matchAndRewrite()
998 int64_t rankRest = dstType.getRank() - rankDiff; in matchAndRewrite()
1009 getI64SubArray(op.offsets(), /*dropFront=*/rankDiff), in matchAndRewrite()
1046 int64_t rankDiff = dstType.getRank() - srcType.getRank(); in matchAndRewrite() local
1047 assert(rankDiff >= 0); in matchAndRewrite()
1048 if (rankDiff != 0) in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 828 unsigned rankDiff = broadcasrSrcRank - extractResultRank; in foldExtractFromBroadcast() local
831 std::next(extractPos.begin(), extractPos.size() - rankDiff)); in foldExtractFromBroadcast()
|