llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Conversion/TorchToMhlo
History
武家伟 ab7aa01b1e
[MHLO] Add torch-to-mhlo e2e support for aten.gather op (#1410) 
* Add torch-to-mhlo e2e support for aten.gather op 

* Add more e2e tests for torch.aten.gather op
 		2022-09-25 22:07:46 +08:00
..
Basic.cpp Fix dynamic shapes type verifications (#1409) 2022-09-23 20:50:29 +08:00
CMakeLists.txt [MLIR][MHLO] Add AtenFrobeniusNormDimOp and add its conversion pattern to MHLO and linalg (#1306) 2022-09-08 10:15:36 +08:00
Gather.cpp [MHLO] Add torch-to-mhlo e2e support for aten.gather op (#1410) 2022-09-25 22:07:46 +08:00
Linear.cpp Fix dynamic shapes type verifications (#1409) 2022-09-23 20:50:29 +08:00
MhloLegalizeUtils.cpp [MHLO] refactor pass configurations (#1315) 2022-09-01 10:36:02 +08:00
MhloLegalizeUtils.h [MHLO] refactor pass configurations (#1315) 2022-09-01 10:36:02 +08:00
Pooling.cpp [MHLO] refactor pass configurations (#1315) 2022-09-01 10:36:02 +08:00
PopulatePatterns.h [MHLO] refactor pass configurations (#1315) 2022-09-01 10:36:02 +08:00
Reduction.cpp [MLIR][MHLO] Add AtenFrobeniusNormDimOp and add its conversion pattern to MHLO and linalg (#1306) 2022-09-08 10:15:36 +08:00
TorchToMhlo.cpp [MHLO] refactor pass configurations (#1315) 2022-09-01 10:36:02 +08:00
ViewLike.cpp [MHLO] refactor pass configurations (#1315) 2022-09-01 10:36:02 +08:00