torch-mlir/lib/Dialect/Torch
Yi Zhang bfc3ee35c6 Import Machine Translation model to MLIR.
This includes the following changes to import MT model into MLIR. There
are still a lot of work to for actual compilation.
- Add `torch.dict<>`, `torch.any`, `torch.number` types
- Add `torch.prim.DictConstruct` op
- Fix `torch.prim.TupleConstruct` op assembly format to include resulting types
2021-08-10 15:22:06 -04:00
..
IR Import Machine Translation model to MLIR. 2021-08-10 15:22:06 -04:00
Transforms Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
CMakeLists.txt Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00