Lines Matching refs:getRank
58 if (sourceType.getRank() != targetType.getRank()) in preservesStaticInformation()
169 int64_t rank = one.getRank(); in joinShapes()
170 if (rank != two.getRank()) in joinShapes()
314 if (*index >= tensorType.getRank()) in verify()
366 if (sliceOp.getType().getRank() == sliceOp.getSourceType().getRank() && in fold()
408 if (tensorType.getRank() != static_cast<int64_t>(getIndices().size())) in verify()
432 auto rank = tensorType.getRank(); in fold()
433 assert(static_cast<int64_t>(indices.size()) == tensorType.getRank() && in fold()
538 if (destType.getRank() != static_cast<int64_t>(getIndices().size())) in verify()
559 reifiedReturnShapes.resize(1, SmallVector<Value>(getType().getRank())); in reifyResultShapes()
561 for (auto dim : llvm::seq<int64_t>(0, getType().getRank())) { in reifyResultShapes()
589 if (getBody().getNumArguments() != resultTy.getRank()) in verifyRegions()
612 auto rank = resultTy.cast<RankedTensorType>().getRank(); in build()
745 return IntegerAttr::get(IndexType::get(getContext()), shapedType.getRank()); in fold()
783 if (shapeSize != resultRankedType.getRank()) in verify()
970 sourceShapedTensorType.getRank() && in inferResultType()
1007 int rankDiff = inferredType.getRank() - desiredResultRank; in inferCanonicalRankReducedResultType()
1155 reifiedReturnShapes[0].reserve(getType().getRank()); in reifyResultShapes()
1210 sliceOp.getType().getRank(), sliceOp.getSourceType(), in matchAndRewrite()
1357 op.getType().getRank(), op.getSourceType(), mixedOffsets, mixedSizes, in operator ()()
1435 unsigned rank = rankedTensorType.getRank(); in createCanonicalRankReducingExtractSliceOp()
1578 reifiedReturnShapes.resize(1, SmallVector<Value>(getType().getRank())); in reifyResultShapes()
1579 for (auto dim : llvm::seq<int64_t>(0, getType().getRank())) { in reifyResultShapes()
1616 insertSliceOp.getSourceType().getRank(), insertSliceOp.getDestType(), in matchAndRewrite()
1736 if (srcType.getRank() != insertSliceOp.getDestType().getRank()) in matchAndRewrite()
1740 for (int64_t i = 0; i < srcType.getRank(); ++i) { in matchAndRewrite()
1788 unsigned rank = rankedTensorType.getRank(); in createCanonicalRankReducingInsertSliceOp()
1829 for (int i = 0, e = sourceType.getRank(); i < e; ++i) { in verify()
1844 unsigned rank = getResult().getType().cast<RankedTensorType>().getRank(); in verifyRegions()
1869 unsigned rank = sourceType.getRank(); in inferResultType()
1909 unsigned rank = sourceType.getRank(); in build()
1941 llvm::SmallBitVector paddedDims(getSourceType().getRank()); in getPaddedDims()
2087 int64_t rank = padOp.getSourceType().getRank(); in matchAndRewrite()
2088 if (outerSliceOp.getSourceType().getRank() != rank) { in matchAndRewrite()
2323 insertSliceOp.getSourceType().getRank(), in matchAndRewrite()