mirror of https://github.com/llvm/torch-mlir
105aad6f57
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. |
||
|---|---|---|
| .. | ||
| fx_importer | ||
| onnx_importer | ||
| compile.py | ||
| lit.local.cfg | ||