torch-mlir/python/torch_mlir/extras
Aart Bik 4147b280ce
[torch-mlir][sparse] add block sparsity to mlir lowering (#2942)
Also note that we are in the process of proposing SparseTensorMetadata
to PyTorch FX graph export (see
https://github.com/pytorch/pytorch/pull/117907). This will hopefully
eventually replace the current data structures in torch-mlir.
2024-02-23 11:57:20 -08:00
..
fx_decomp_util.py add support for decomposition (#2879) 2024-02-14 21:00:52 -08:00
fx_importer.py [torch-mlir][sparse] add block sparsity to mlir lowering (#2942) 2024-02-23 11:57:20 -08:00
onnx_importer.py [onnx] Drop `ConstantOfShape` logic form importer, fix torch lowering (#2930) 2024-02-21 21:34:43 -08:00