tvm.sparse.condense
- tvm.sparse.condense(indptr_nd, indices_nd, t, g)[source]
Condense sparse matrix in CSR format to (t x 1) tiles, and group g tiles together.
- indptrNDArray
The indptr array of CSR format.
- indicesNDArray
The indices array of CSR format.
- tint
The tile size.
- gint
The group size.
- Tuple[NDArray]
The pair of (group_indptr, tile_indices, mask).