Searched defs:nnz (Results 1 – 5 of 5) sorted by relevance
| /llvm-project/mlir/include/mlir/ExecutionEngine/SparseTensor/ |
| H A D | Storage.h | 274 uint64_t nnz = values.size(); in getCoordinatesBuffer() local
357 uint64_t nnz = values.size(); in sortInPlace() local
|
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | CudaRuntimeWrappers.cpp | 630 CUSPARSE_REPORT_IF_ERROR(cusparseCreateCsr(&mat, rows, cols, nnz, rowPos, in mgpuCreateCsc() argument
590 mgpuCreateCoo(intptr_t rows,intptr_t cols,intptr_t nnz,void * rowIdxs,void * colIdxs,void * values,int32_t itp,int32_t dtp,CUstream) mgpuCreateCoo() argument
604 mgpuCreateCooAoS(intptr_t rows,intptr_t cols,intptr_t nnz,void * idxs,void * values,int32_t itp,int32_t dtp,CUstream) mgpuCreateCooAoS() argument
616 mgpuCreateCsr(intptr_t rows,intptr_t cols,intptr_t nnz,void * rowPos,void * colIdxs,void * values,int32_t ptp,int32_t itp,int32_t dtp,CUstream) mgpuCreateCsr() argument
838 int64_t *nnz = reinterpret_cast<int64_t *>(n); mgpuSpMatGetSize() local
[all...] |
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorRewriting.cpp | 897 Value nnz = rewriter.create<NumberOfEntriesOp>(loc, srcTensor); matchAndRewrite() local
1005 Value nnz = rewriter.create<NumberOfEntriesOp>(loc, srcTensor); matchAndRewrite() local
1502 Value nnz = rewriter.create<NumberOfEntriesOp>(loc, src); matchAndRewrite() local
|
| H A D | SparseGPUCodegen.cpp | 877 Value nnz = sizes->getResult(2); in rewriteSpGEMM() local
|
| H A D | SparseTensorCodegen.cpp | 677 auto nnz = genValMemSize(rewriter, op.getLoc(), adaptor.getInputCoo()); matchAndRewrite() local
|