mirror of https://github.com/llvm/torch-mlir
1f00f95d2e
Implements a python package for taking a `torch.fx.GraphModule` and turning it into MLIR in the `torch` dialect that can then be further compiled by `npcomp`. This is a WIP, so the coverage of PyTorch operations is very small. |
||
---|---|---|
.. | ||
torchfx | ||
__init__.py | ||
resnet_inference.ipynb | ||
torchscript_resnet18_e2e.py |