tvm.sparse.specialize_buffer
- tvm.sparse.specialize_buffer(mod: tvm.ir.module.IRModule, buf_name: str, idx_map: Union[Callable, tvm.tir.function.IndexMap])[source]
Specialize a buffer in an IRModule with given buffer name and index map function.
- modIRModule
The IRModule we perform the specialization.
- buf_namestr
The name of the buffer to specialize.
- idx_mapIndexMap
The index map.
- IRModule
The new IRModule.