torch-mlir/lib/Dialect/Torch
Qiang Fu f7c7bb800c
Add non-default dtype support for a few elementwise math ops. (#687)
* fix type inference
* fix Torch2Linalg conversion
* add test cases
2022-03-23 13:35:43 -07:00
..
IR Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00
Transforms Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00
Utils Avoid `using` the `torch_upstream` namespace. 2022-03-15 17:24:17 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00