llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/frontends/pytorch/e2e_testing/torchscript
History
Sean Silva 902c2e579b Add resnet inference jupyter notebook. 
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.
 		2021-08-09 14:34:43 -07:00
..
basic.py Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
batchnorm.py Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
conv.py Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
elementwise.py Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
main.py Add resnet inference jupyter notebook. 2021-08-09 14:34:43 -07:00
mlp.py Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
quantized_models.py Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
vision_models.py Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
xfail_sets.py Triage e2e IREE failures for npcetest. 2021-08-05 12:14:58 -07:00