torch-mlir/python/torch_mlir/extras
Aart Bik f72770a725
[torch-mlir][sparse] replace ad-hoc mechanism with proper FX export (#3648)
Now that the PyDev feature request pytorch/pytorch#117188 has been
completed, we can remove all the ad-hoc code that propagates sparsity
metadata and replace it with the built-int PyDev metadata for sparse
tensors. This removes a lot of code and also ensures sparsity is
consistent with the torch.sparse package for all cases.
2024-08-20 09:56:21 -07:00
..
fx_decomp_util.py [FxImporter] small fixes for fx importer compatibility issues between different pytorch versions (#3577) 2024-08-01 10:52:41 +08:00
fx_importer.py [torch-mlir][sparse] replace ad-hoc mechanism with proper FX export (#3648) 2024-08-20 09:56:21 -07:00
onnx_importer.py Bump Onnx Version to 1.16.1 (#3515) 2024-07-01 22:15:45 +05:30