Searched refs:elemTp (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | CodegenUtils.cpp | 102 PrimaryType mlir::sparse_tensor::primaryTypeEncoding(Type elemTp) { in primaryTypeEncoding() argument
103 if (elemTp.isF64()) in primaryTypeEncoding()
105 if (elemTp.isF32()) in primaryTypeEncoding()
107 if (elemTp.isF16()) in primaryTypeEncoding()
109 if (elemTp.isBF16()) in primaryTypeEncoding()
111 if (elemTp.isInteger(64)) in primaryTypeEncoding()
113 if (elemTp.isInteger(32)) in primaryTypeEncoding()
115 if (elemTp.isInteger(16)) in primaryTypeEncoding()
117 if (elemTp.isInteger(8)) in primaryTypeEncoding()
119 if (auto complexTp = elemTp.dyn_cast<ComplexType>()) { in primaryTypeEncoding()
[all …]
|
| H A D | CodegenUtils.h | 63 PrimaryType primaryTypeEncoding(Type elemTp);
69 StringRef primaryTypeFunctionSuffix(Type elemTp);
183 Type elemTp) { in constantPrimaryTypeEncoding() argument
185 static_cast<uint32_t>(primaryTypeEncoding(elemTp))); in constantPrimaryTypeEncoding()
|
| H A D | SparseTensorConversion.cpp | 225 Type elemTp = stp.getElementType(); in newParams() local
330 Type elemTp = tensorTp.getElementType(); in allocDenseTensor() local
332 auto memTp = MemRefType::get(shape, elemTp); in allocDenseTensor()
339 Value zero = constantZero(builder, loc, elemTp); in allocDenseTensor()
469 Type elemTp = srcTp.getElementType(); in genSparse2SparseReshape() local
470 assert(elemTp == dstTp.getElementType() && in genSparse2SparseReshape()
492 Value elemPtr = genAllocaScalar(rewriter, loc, elemTp); in genSparse2SparseReshape()
512 genDelCOOCall(rewriter, op, elemTp, coo); in genSparse2SparseReshape()
513 genDelCOOCall(rewriter, op, elemTp, iter); in genSparse2SparseReshape()
738 Type elemTp = dstTensorTp.getElementType(); in matchAndRewrite() local
[all …]
|