llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/examples
History
Sean Silva 075464fa74 Add a new `torch_mlir.compile` method. 
This makes it much easier to convert models and hides all the
ClassAnnotator complexity.

This also adds a new example `torchscript_resnet18_all_output_types.py`
which shows the ResNet18 IR for all output types.

Also,

- This moves `run_pipeline_with_repro_report` to
  `torch_mlir.compiler_utils`.
 		2022-04-20 10:06:01 -07:00
..
eager_mode.py Eager mode description in the README and small example and ResNet18 example. (#707) 2022-03-28 23:54:06 -05:00
eager_mode_resnet18.py Eager mode description in the README and small example and ResNet18 example. (#707) 2022-03-28 23:54:06 -05:00
torchscript_resnet18.py Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00
torchscript_resnet18_all_output_types.py Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00
torchscript_resnet_inference.ipynb Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00