Searched refs:dstTp (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
397 unsigned dstRank = dstTp.getRank(); in translateIndices()
402 ArrayRef<int64_t> shape = isExpand ? srcTp.getShape() : dstTp.getShape(); in translateIndices()
462 RankedTensorType dstTp, RankedTensorType srcTp) { in genSparse2SparseReshape() argument
464 auto encDst = getSparseTensorEncoding(dstTp); in genSparse2SparseReshape()
468 unsigned dstRank = dstTp.getRank(); in genSparse2SparseReshape()
470 assert(elemTp == dstTp.getElementType() && in genSparse2SparseReshape()
485 sizesFromPtr(rewriter, sizes, op, encDst, dstTp, src); in genSparse2SparseReshape()
486 newParams(rewriter, params, op, dstTp, encDst, Action::kEmptyCOO, sizes); in genSparse2SparseReshape()
504 translateIndices(loc, rewriter, reassociation, dstTp, srcTp, dstIdx, srcIdx); in genSparse2SparseReshape()
|
| H A D | Sparsification.cpp | 1788 auto dstTp = RankedTensorType::get(srcTp.getShape(), in resolveCycle() local
1790 auto convert = rewriter.create<ConvertOp>(tval.getLoc(), dstTp, tval); in resolveCycle()
|