tvm.sparse.csf_to_ell3d
- tvm.sparse.csf_to_ell3d(csf_indptr_0, csf_indices_0, csf_indptr_1, csf_indices_1, nnz_rows_bkt, nnz_cols_bkt)[source]
Convert CSF format to composable ELL format in 3-dimensional setting (HeteroGraphs).
- csf_indptr_0NDArray
Level 0 indptr array in CSF format.
- csf_indices_0NDArray
Level 0 indices array in CSF format.
- csf_indptr_1NDArray
Level 1 indptr array in CSF format.
- csf_indices_1NDArray
Level 1 indices array in CSF format.
- num_rows_bktList[int]
Number of non-zero rows bucket.
- nnz_cols_bktList[int]
Number of non-zero columns bucket.
- Tuple[List[NDArray]]
(indptr, row_indices, col_indices, mask) Each one is a list of NDArray, with length #rels.