Lines Matching refs:getRank
128 elementVectorRank += elementVectorType.getRank(); in getTransferMinorIdentityMap()
131 if (shapedType.getRank() == 0 && in getTransferMinorIdentityMap()
137 shapedType.getRank(), vectorType.getRank() - elementVectorRank, in getTransferMinorIdentityMap()
349 if (getSourceVectorType().getRank() == 1 && !isReducedDim(0)) in fold()
397 int64_t rank = getVectorType().getRank(); in verify()
626 if (dimPair.first < 0 || dimPair.first >= lhsType.getRank() || in verifyDimMap()
627 dimPair.second < 0 || dimPair.second >= rhsType.getRank() || in verifyDimMap()
651 for (int64_t i = 0, e = lhsType.getRank(); i < e; ++i) { in verifyOutputShape()
658 for (int64_t i = 0, e = rhsType.getRank(); i < e; ++i) { in verifyOutputShape()
690 for (unsigned idx = 0, e = v.getRank(); idx < e; ++idx) { in verifyOutputShape()
966 if (vectorType.getRank() == 0) { in verify()
971 if (vectorType.getRank() != 1) in verify()
1027 if (static_cast<int64_t>(op.getPosition().size()) == vectorType.getRank()) { in inferReturnTypes()
1031 std::min<size_t>(op.getPosition().size(), vectorType.getRank() - 1); in inferReturnTypes()
1053 if (positionAttr.size() > static_cast<unsigned>(getVectorType().getRank())) in verify()
1201 : extractOp(e), vectorRank(extractOp.getVectorType().getRank()), in ExtractFromInsertTransposeChainState()
1322 auto getRank = [](Type type) { in foldExtractFromBroadcast() local
1323 return type.isa<VectorType>() ? type.cast<VectorType>().getRank() : 0; in foldExtractFromBroadcast()
1325 unsigned broadcastSrcRank = getRank(source.getType()); in foldExtractFromBroadcast()
1326 unsigned extractResultRank = getRank(extractOp.getType()); in foldExtractFromBroadcast()
1359 ? extractOp.getType().cast<VectorType>().getRank() in foldExtractFromShapeCast()
1361 if (destinationRank > shapeCastOp.getSourceVectorType().getRank()) in foldExtractFromShapeCast()
1390 shapeCastOp.getSourceVectorType().getRank() - destinationRank; in foldExtractFromShapeCast()
1430 destinationRank = vecType.getRank(); in foldExtractFromExtractStrided()
1434 extractStridedSliceOp.getVectorType().getRank() - sliceOffsets.size()) in foldExtractFromExtractStrided()
1451 ? op.getType().cast<VectorType>().getRank() in foldExtractStridedOpFromInsertChain()
1455 int64_t insertRankDiff = insertOp.getDestVectorType().getRank() - in foldExtractStridedOpFromInsertChain()
1456 insertOp.getSourceVectorType().getRank(); in foldExtractStridedOpFromInsertChain()
1457 if (destinationRank > insertOp.getSourceVectorType().getRank()) in foldExtractStridedOpFromInsertChain()
1490 insertOp.getSourceVectorType().getRank() - destinationRank; in foldExtractStridedOpFromInsertChain()
1545 auto getRank = [](Type type) { in matchAndRewrite() local
1546 return type.isa<VectorType>() ? type.cast<VectorType>().getRank() : 0; in matchAndRewrite()
1548 unsigned broadcastSrcRank = getRank(source.getType()); in matchAndRewrite()
1549 unsigned extractResultRank = getRank(extractOp.getType()); in matchAndRewrite()
1621 if (getSourceVectorType().getRank() != getResultType().getRank()) in verify()
1624 for (unsigned i = 0, e = getSourceVectorType().getRank(); i < e; ++i) { in verify()
1648 for (unsigned i = 0, e = getSourceVectorType().getRank(); i < e; i++) { in getMultiplicity()
1660 for (unsigned i = 0, e = op.getSourceVectorType().getRank(); i < e; i++) { in calculateImplicitMap()
1665 auto map = AffineMap::get(op.getSourceVectorType().getRank(), 0, perm, in calculateImplicitMap()
1696 int64_t srcRank = srcVectorType.getRank(); in isBroadcastableTo()
1697 int64_t dstRank = dstVectorType.getRank(); in isBroadcastableTo()
1786 int64_t resRank = resultType.getRank(); in verify()
1787 int64_t v1Rank = v1Type.getRank(); in verify()
1788 int64_t v2Rank = v2Type.getRank(); in verify()
1826 shape.reserve(v1Type.getRank()); in inferReturnTypes()
1861 if (lhsType.getRank() != 1) in fold()
1921 if (dstVectorType.getRank() == 0) { in verify()
1926 if (dstVectorType.getRank() != 1) in verify()
1981 if (positionAttr.size() > static_cast<unsigned>(destVectorType.getRank())) in verify()
1986 (static_cast<unsigned>(srcVectorType.getRank()) + positionAttr.size() != in verify()
1987 static_cast<unsigned>(destVectorType.getRank()))) in verify()
1991 (positionAttr.size() != static_cast<unsigned>(destVectorType.getRank()))) in verify()
2068 if (getSourceVectorType().getRank() != getResultType().getRank()) in verify()
2071 for (unsigned i = 0, e = getResultType().getRank(); i < e; i++) { in verify()
2198 if (offsets.size() != static_cast<unsigned>(destVectorType.getRank())) in verify()
2201 if (strides.size() != static_cast<unsigned>(sourceVectorType.getRank())) in verify()
2203 if (sourceVectorType.getRank() > destVectorType.getRank()) in verify()
2361 if (vLHS.getRank() != 1) in verify()
2366 if (vRHS.getRank() != 1) in verify()
2368 if (vRES.getRank() != 2) in verify()
2376 if (vRES.getRank() != 1) in verify()
2406 if (inputVectorType.getRank() != inputShapeRank + numFixedVectorSizes) in verify()
2409 if (outputVectorType.getRank() != outputShapeRank + numFixedVectorSizes) in verify()
2415 unsigned inputVectorRank = inputVectorType.getRank(); in verify()
2423 unsigned outputVectorRank = outputVectorType.getRank(); in verify()
2468 shape.reserve(vectorType.getRank()); in inferStridedSliceOpResultType()
2548 if (op.getVectorType().getRank() != in foldExtractStridedOpFromInsertChain()
2549 insertOp.getSourceVectorType().getRank()) in foldExtractStridedOpFromInsertChain()
2707 unsigned srcRank = srcVecType ? srcVecType.getRank() : 0; in matchAndRewrite()
2709 unsigned dstRank = dstVecType.getRank(); in matchAndRewrite()
2878 unsigned sourceVecEltRank = vectorElementType.getRank(); in verifyTransferOp()
2879 unsigned resultVecRank = vectorType.getRank(); in verifyTransferOp()
2894 vectorType.getRank() == 0 ? 1 : vectorType.getShape().back(); in verifyTransferOp()
2903 if (permutationMap.getNumResults() != vectorType.getRank()) in verifyTransferOp()
2918 if (permutationMap.getNumInputs() != shapedType.getRank()) in verifyTransferOp()
3034 if (static_cast<int64_t>(getIndices().size()) != shapedType.getRank()) in verify()
3035 return emitOpError("requires ") << shapedType.getRank() << " indices"; in verify()
3251 extractOp.getSourceType().getRank() - extractOp.getType().getRank(); in matchAndRewrite()
3252 int64_t vectorRank = xferOp.getVectorType().getRank(); in matchAndRewrite()
3491 if (llvm::size(getIndices()) != shapedType.getRank()) in verify()
3492 return emitOpError("requires ") << shapedType.getRank() << " indices"; in verify()
3723 if (xferOp.getVectorType().getRank() != xferOp.getShapedType().getRank()) in matchAndRewrite()
3753 insertOp.getType().getRank() - insertOp.getSourceType().getRank(); in matchAndRewrite()
3754 int64_t vectorRank = xferOp.getVectorType().getRank(); in matchAndRewrite()
3817 if (insertOp.getSourceType().getRank() != transferOp.getTransferRank()) { in matchAndRewrite()
3913 if (llvm::size(getIndices()) != memRefTy.getRank()) in verify()
3914 return emitOpError("requires ") << memRefTy.getRank() << " indices"; in verify()
3946 if (llvm::size(getIndices()) != memRefTy.getRank()) in verify()
3947 return emitOpError("requires ") << memRefTy.getRank() << " indices"; in verify()
3968 if (llvm::size(getIndices()) != memType.getRank()) in verify()
3969 return emitOpError("requires ") << memType.getRank() << " indices"; in verify()
4021 if (llvm::size(getIndices()) != memType.getRank()) in verify()
4022 return emitOpError("requires ") << memType.getRank() << " indices"; in verify()
4072 if (llvm::size(getIndices()) != memType.getRank()) in verify()
4073 return emitOpError("requires ") << memType.getRank() << " indices"; in verify()
4120 if (llvm::size(getIndices()) != memType.getRank()) in verify()
4121 return emitOpError("requires ") << memType.getRank() << " indices"; in verify()
4166 if (llvm::size(getIndices()) != memType.getRank()) in verify()
4167 return emitOpError("requires ") << memType.getRank() << " indices"; in verify()
4213 if (llvm::size(getIndices()) != memType.getRank()) in verify()
4214 return emitOpError("requires ") << memType.getRank() << " indices"; in verify()
4300 unsigned sourceRank = sourceVectorType.getRank(); in verifyVectorShapeCast()
4301 unsigned resultRank = resultVectorType.getRank(); in verifyVectorShapeCast()
4336 if (srcType.getRank() < resultType.getRank()) { in fold()
4339 } else if (srcType.getRank() > resultType.getRank()) { in fold()
4433 for (int64_t i = 0, e = sourceVectorType.getRank() - 1; i < e; i++) { in verify()
4444 if (sourceVectorType.getRank() == 0) { in verify()
4555 SmallVector<int64_t, 4> transposedShape(vt.getRank()); in build()
4588 int64_t rank = resultType.getRank(); in verify()
4589 if (vectorType.getRank() != rank) in verify()
4715 if (resultType.getRank() == 0) { in verify()
4725 if (static_cast<int64_t>(getMaskDimSizes().size()) != resultType.getRank()) in verify()
4764 if (vectorType.getRank() == 0) { in verify()
4769 getResult().getType().cast<VectorType>().getRank()) { in verify()
4814 maskDimSizes.assign(createMaskOp.getType().getRank(), 0); in matchAndRewrite()
4842 int64_t srcRank = srcType.getRank(); in verify()
4849 int64_t initialValueRank = initialType.getRank(); in verify()
5027 if (expandedVecType.getRank() != distributedVecType.getRank() || in verifyDistributedType()
5032 for (int64_t i = 0, e = expandedVecType.getRank(); i < e; i++) { in verifyDistributedType()