torch-mlir/lib/Conversion/TorchToMhlo
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
..
Basic.cpp Add decomposition for aten.roll (#1170) 2022-08-24 08:36:05 +08:00
CMakeLists.txt [MHLO] Init end to end unit tests (#1223) 2022-08-23 16:47:21 +08:00
Gather.cpp [MHLO] Init MHLO pooling-like op conversion (#1141) 2022-08-04 12:34:22 +08:00
Linear.cpp build: update llvm tag to 2dde4ba6 (#1229) 2022-08-15 23:54:45 -07:00
MhloLegalizeUtils.cpp [MHLO] support non-constant torch scalar in BasicOps (#1134) 2022-08-03 08:16:31 +08:00
MhloLegalizeUtils.h [MHLO] support non-constant torch scalar in BasicOps (#1134) 2022-08-03 08:16:31 +08:00
Pooling.cpp llvm: update tag to 061e0189 (#1180) 2022-08-08 20:17:35 -07:00
PopulatePatterns.h [MHLO] Init MHLO pooling-like op conversion (#1141) 2022-08-04 12:34:22 +08:00
Reduction.cpp [MHLO] Init end to end unit tests (#1223) 2022-08-23 16:47:21 +08:00
TorchToMhlo.cpp remove torch dialect from legal list (#1192) 2022-08-11 09:22:41 +08:00
ViewLike.cpp [MHLO] Init end to end unit tests (#1223) 2022-08-23 16:47:21 +08:00