torch-mlir/lib/Dialect/Torch
Liam Fitzpatrick ccfdfd1b80 Refine static shapes for conv2d and maxpool2d 2022-01-03 11:09:23 -06:00
..
IR [TORCH][MLIR] Fold trivial cases of `aten.to.dtype` and `aten.view` op 2021-12-24 13:32:34 +05:30
Transforms Refine static shapes for conv2d and maxpool2d 2022-01-03 11:09:23 -06:00
Utils Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00