Home
last modified time | relevance | path

Searched defs:memTp (Results 1 – 4 of 4) sorted by relevance

/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DCodegenUtils.cpp363 auto memTp = MemRefType::get({sz}, tp); in genAlloca() local
371 auto memTp = MemRefType::get({ShapedType::kDynamic}, tp); in genAlloca() local
397 auto memTp = MemRefType::get(shape, elemTp); in allocDenseTensor() local
511 const auto memTp = cast<MemRefType>(mem.getType()); in loadAll() local
533 const auto memTp in storeAll() local
612 auto memTp = MemRefType::get({ShapedType::kDynamic}, indexTp); genReader() local
[all...]
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseGPUCodegen.cpp115 MemRefType memTp = cast<MemRefType>(mem.getType()); in genHostRegisterMemref() local
151 auto memTp = MemRefType::get(shape, elemTp); in genAllocMemRef() local
166 const auto memTp = MemRefType::get({ShapedType::kDynamic}, type); in genHostBuffer() local
173 const auto memTp = MemRefType::get({ShapedType::kDynamic}, type); in genAllocBuffer() local
H A DSparseTensorCodegen.cpp420 auto memTp = llvm::cast<MemRefType>(mem.getType()); genSliceToSize() local
924 const auto memTp = MemRefType::get({ShapedType::kDynamic}, t); matchAndRewrite() local
H A DSparseTensorConversion.cpp139 auto memTp = MemRefType::get({ShapedType::kDynamic}, tp); in genAlloc() local