Searched refs:denseTp (Results 1 – 2 of 2) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorRewriting.cpp | 216 auto denseTp = in matchAndRewrite() local
218 auto convert = rewriter.create<ConvertOp>(loc, denseTp, op.getSrc()); in matchAndRewrite()
224 auto denseTp = in matchAndRewrite() local
226 auto reshape = rewriter.create<ReshapeOp>(loc, denseTp, op.getSrc(), in matchAndRewrite()
|
| H A D | Sparsification.cpp | 471 linalg::GenericOp op, MemRefType denseTp, in genOutputBuffer() argument
484 Value init = builder.create<bufferization::ToMemrefOp>(loc, denseTp, tensor); in genOutputBuffer()
486 Value zero = constantZero(builder, loc, denseTp.getElementType()); in genOutputBuffer()
543 auto denseTp = MemRefType::get(shape, elementType); in genBuffers() local
546 builder.create<bufferization::ToMemrefOp>(loc, denseTp, t->get()); in genBuffers()
549 genOutputBuffer(codegen, builder, op, denseTp, args); in genBuffers()
|