torch-mlir

Commit Graph

Author	SHA1	Message	Date
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
Sean Silva	1960ba76fb	Remove "e2e" name from `examples/torchscript_resnet18_e2e.py` That was back from an earlier stage in the project when e2e was a big deal because we didn't have anything working e2e yet :)	2022-03-28 18:26:54 +00:00

Author

SHA1

Message

Date

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

Sean Silva

1960ba76fb

Remove "e2e" name from `examples/torchscript_resnet18_e2e.py`

That was back from an earlier stage in the project when e2e was a big
deal because we didn't have anything working e2e yet :)

2022-03-28 18:26:54 +00:00

2 Commits (81ee5bb58c57cb88ea69005db8ce655bc8b3bb78)