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.py | ||
onnx_importer.py |