Lines Matching refs:getRank
209 if (vectorType.getRank() > 0) { in buildVectorWrite()
217 SmallVector<Value> indices(linalgOp.getRank(outputOperand), in buildVectorWrite()
493 if (readValue.getType().cast<VectorType>().getRank() == 0) in vectorizeAsLinalgGeneric()
637 SmallVector<Value> indices(srcType.getRank(), zero); in vectorizeCopy()
641 rewriter.getMultiDimIdentityMap(srcType.getRank())); in vectorizeCopy()
642 if (readValue.getType().cast<VectorType>().getRank() == 0) { in vectorizeCopy()
648 rewriter.getMultiDimIdentityMap(srcType.getRank())); in vectorizeCopy()
711 for (unsigned i = 0; i < sourceType.getRank(); ++i) { in tryVectorizeCopy()
740 vecType.getRank(), in tryVectorizeCopy()
822 SmallVector<bool> inBounds(xferOp.getVectorType().getRank(), false); in rewriteUser()
899 SmallVector<bool> inBounds(xferOp.getVectorType().getRank(), false); in rewriteUser()
933 if (t1.getRank() != t2.getRank()) in hasSameTensorSize()
938 for (unsigned i = 0; i < t1.getRank(); ++i) { in hasSameTensorSize()
958 assert(static_cast<size_t>(t1.getRank()) == in hasSameTensorSize()
960 assert(static_cast<size_t>(t2.getRank()) == in hasSameTensorSize()
963 for (unsigned i = 0; i < t1.getRank(); ++i) { in hasSameTensorSize()
1045 unsigned vecRank = vecType.getRank(); in rewriteUser()
1046 unsigned tensorRank = insertOp.getType().getRank(); in rewriteUser()
1332 if (lhsShapedType.getRank() != 3 || in Conv1DNwcGenerator()
1333 (rhsShapedType.getRank() != 2 && rhsShapedType.getRank() != 3) || in Conv1DNwcGenerator()
1334 resShapedType.getRank() != 3) in Conv1DNwcGenerator()