torch-mlir/lib/Dialect/Torch
Tanyo Kwok 3d0e18bbe7
Add decomposition for aten.roll (#1170)
* Add decomposition for aten.roll

* add e2e unittest

* refine type of torch.roll

* fix aten::cat output type
2022-08-24 08:36:05 +08:00
..
IR Generalize canonicalization pattern for more aten.sub/div/mul/add op (#1209) 2022-08-16 13:24:08 +08:00
Transforms Add decomposition for aten.roll (#1170) 2022-08-24 08:36:05 +08:00
Utils importer: add initial support for loading Float16 tensors (#1169) 2022-08-08 12:37:31 +08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00