Searched refs:srcTp (Results 1 – 2 of 2) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorConversion.cpp | 395 TensorType dstTp, TensorType srcTp, Value dstIdx, in translateIndices() argument
398 unsigned srcRank = srcTp.getRank(); in translateIndices()
402 ArrayRef<int64_t> shape = isExpand ? srcTp.getShape() : dstTp.getShape(); in translateIndices()
462 RankedTensorType dstTp, RankedTensorType srcTp) { in genSparse2SparseReshape() argument
465 auto encSrc = getSparseTensorEncoding(srcTp); in genSparse2SparseReshape()
467 unsigned srcRank = srcTp.getRank(); in genSparse2SparseReshape()
469 Type elemTp = srcTp.getElementType(); in genSparse2SparseReshape()
478 sizesFromPtr(rewriter, sizes, op, noPerm, srcTp, src); in genSparse2SparseReshape()
479 newParams(rewriter, params, op, srcTp, noPerm, Action::kToIterator, sizes, in genSparse2SparseReshape()
504 translateIndices(loc, rewriter, reassociation, dstTp, srcTp, dstIdx, srcIdx); in genSparse2SparseReshape()
|
| H A D | Sparsification.cpp | 1783 auto srcTp = tval.getType().cast<RankedTensorType>(); in resolveCycle() local
1788 auto dstTp = RankedTensorType::get(srcTp.getShape(), in resolveCycle()
1789 srcTp.getElementType(), dstEnc); in resolveCycle()
|