torch-mlir/examples
Ramiro Leal-Cavazos 1f00f95d2e
WIP implementation of torchfx (#304)
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.
2021-09-22 10:27:55 -07:00
..
torchfx WIP implementation of torchfx (#304) 2021-09-22 10:27:55 -07:00
__init__.py Remove frontends/pytorch directory. 2021-09-17 09:27:49 -07:00
resnet_inference.ipynb Remove frontends/pytorch directory. 2021-09-17 09:27:49 -07:00
torchscript_resnet18_e2e.py Remove frontends/pytorch directory. 2021-09-17 09:27:49 -07:00