tvm.sparse.format_decompose

tvm.sparse.format_decompose(mod: tvm.ir.module.IRModule, composable_formats: List[tvm.sparse.format.FormatRewriteRule], include_format_rewrite_blks: bool = True)

Rewrite the sparse format of sparse buffers in the TIR scripts.

modIRModule

The IRModule to lower.

composable_formatsList[FormatRewriteRule]

Composable formats is a list of rewrite rules.

include_format_rewrite_blksbool

Whether to include format rewrite blocks in the output.