mirror of https://github.com/llvm/torch-mlir
902c2e579b
This takes the example from torchscript_resnet18_e2e.py and puts it into a slightly cleaned up notebook form. It's still a little rough around the edges. Areas for improvement: - Installation / setup. - API usability. Also, - Add `npcomp-backend-to-iree-frontend-pipeline` since we will be adding more stuff there. - Slight cleanups. |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| cos_e2e.py | ||
| div_inplace_e2e.py | ||
| mm_e2e.py | ||
| mul_maximum_e2e.py | ||
| resnet_inference.ipynb | ||
| tanh_out_e2e.py | ||
| test_utils.py | ||
| torchscript_mm_e2e.py | ||
| torchscript_resnet18_e2e.py | ||
| torchscript_tanh_e2e.py | ||
| torchscript_tanh_e2e_iree.py | ||