torch-mlir/python/torch_mlir/extras
Aart Bik 105aad6f57
[torch-mlir] provide FX traced graph importer for sparse tensors (#2817)
Note that we are waiting for actual FX traced graph support for sparse
tensors. For details see

https://github.com/pytorch/pytorch/issues/117188

Until then, however, we provide this clever importer that builds the FX
traced graph for for the dense case and then puts a sparse annotation
back on the parameters.

With import test.
2024-01-30 21:22:12 -08:00
..
fx_importer.py [torch-mlir] provide FX traced graph importer for sparse tensors (#2817) 2024-01-30 21:22:12 -08:00
onnx_importer.py Fix onnx importer to treat Constant values as static (#2780) 2024-01-22 13:00:05 -08:00