Searched refs:rankDiff (Results 1 – 6 of 6) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorInsertExtractStridedSliceRewritePatterns.cpp | 69 int64_t rankDiff = dstType.getRank() - srcType.getRank(); in matchAndRewrite() local
70 assert(rankDiff >= 0); in matchAndRewrite()
71 if (rankDiff == 0) in matchAndRewrite()
74 int64_t rankRest = dstType.getRank() - rankDiff; in matchAndRewrite()
86 getI64SubArray(op.getOffsets(), /*dropFront=*/rankDiff), in matchAndRewrite()
|
| H A D | VectorTransforms.cpp | 1109 int64_t rankDiff = in matchAndRewrite() local
1115 broadcast.getVectorType().getDimSize(rankDiff + dim.index())) { in matchAndRewrite()
1120 rewriter.getAffineDimExpr(dim.index() + rankDiff)); in matchAndRewrite()
1130 for (int64_t i = 0; i < rankDiff; ++i) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/ShapeToStandard/ |
| H A D | ShapeToStandard.cpp | 84 Value rankDiff = std::get<1>(tup); in getBroadcastedDim() local
86 outputDimension, rankDiff); in getBroadcastedDim()
103 loc, indexTy, outputDimension, rankDiff); in getBroadcastedDim()
285 Value shape, rankDiff; in matchAndRewrite() local
286 std::tie(shape, rankDiff) = tup; in matchAndRewrite()
288 loc, arith::CmpIPredicate::ult, iv, rankDiff); in matchAndRewrite()
300 b.create<arith::SubIOp>(loc, indexTy, iv, rankDiff); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/ |
| H A D | TosaOps.cpp | 888 auto rankDiff = outShape.size() - shape.getRank(); in REDUCE_SHAPE_INFER() local
891 auto dim1 = outShape[i + rankDiff]; in REDUCE_SHAPE_INFER()
902 outShape[i + rankDiff] = resolvedDim; in REDUCE_SHAPE_INFER()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 1337 unsigned rankDiff = broadcastSrcRank - extractResultRank; in foldExtractFromBroadcast() local
1339 std::next(extractPos.begin(), extractPos.size() - rankDiff)); in foldExtractFromBroadcast()
2710 unsigned rankDiff = dstRank - srcRank; in matchAndRewrite() local
2716 if (srcVecType.getDimSize(i) != dstVecType.getDimSize(i + rankDiff)) { in matchAndRewrite()
2730 getI64SubArray(op.getOffsets(), /* dropFront=*/rankDiff), in matchAndRewrite()
2731 getI64SubArray(op.getSizes(), /* dropFront=*/rankDiff), in matchAndRewrite()
2732 getI64SubArray(op.getStrides(), /* dropFront=*/rankDiff)); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 1007 int rankDiff = inferredType.getRank() - desiredResultRank; in inferCanonicalRankReducedResultType() local
1008 if (rankDiff > 0) { in inferCanonicalRankReducedResultType()
1011 getPositionsOfShapeOne(rankDiff, shape); in inferCanonicalRankReducedResultType()
|