Home
last modified time | relevance | path

Searched refs:dstRank (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorInsertExtractStridedSliceRewritePatterns.cpp125 int64_t dstRank = dstType.getRank(); in matchAndRewrite() local
126 assert(dstRank >= srcRank); in matchAndRewrite()
127 if (dstRank != srcRank) in matchAndRewrite()
H A DVectorTransforms.cpp270 int64_t dstRank = dstType.getRank(); in matchAndRewrite() local
273 if (srcRank <= 1 && dstRank == 1) { in matchAndRewrite()
292 if (srcRank < dstRank) { in matchAndRewrite()
307 assert(srcRank == dstRank); in matchAndRewrite()
309 for (int64_t r = 0; r < dstRank; r++) in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorConversion.cpp397 unsigned dstRank = dstTp.getRank(); in translateIndices() local
401 bool isExpand = srcRank > dstRank; in translateIndices()
442 assert((isExpand && i == dstRank) || (!isExpand && i == srcRank)); in translateIndices()
468 unsigned dstRank = dstTp.getRank(); in genSparse2SparseReshape() local
491 Value dstIdx = genAlloca(rewriter, loc, dstRank, rewriter.getIndexType()); in genSparse2SparseReshape()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp1697 int64_t dstRank = dstVectorType.getRank(); in isBroadcastableTo() local
1698 if (srcRank > dstRank) in isBroadcastableTo()
1702 int64_t lead = dstRank - srcRank; in isBroadcastableTo()
2709 unsigned dstRank = dstVecType.getRank(); in matchAndRewrite() local
2710 unsigned rankDiff = dstRank - srcRank; in matchAndRewrite()