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/TorchToStablehlo
History
Xinyu Yang 431d98b405
[Stablehlo] Add lowering of GridSampler Op (#3084) 
Inspired by PyTorch decompositions.py.
See
ec58f1f74e/torch/_decomp/decompositions.py (L3923-L4086)
Only support paddingMode=0 or 1 and interpolationMode=0 or 1
 		2024-06-07 16:06:07 +08:00
..
Basic.cpp [Stablehlo] support uint8 (#3367) 2024-06-04 09:04:59 +08:00
CMakeLists.txt [Stablehlo] lowering aten.randn & aten.normal_functional to mhlo.rng … (#3328) 2024-05-11 15:33:37 +08:00
GatherScatter.cpp [Stablehlo] Add lowering of GridSampler Op (#3084) 2024-06-07 16:06:07 +08:00
Linear.cpp [Torch]Support conv_transpose1d and conv_transpose3d (#3286) 2024-06-03 15:11:12 +08:00
Pooling.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
PopulatePatterns.h [Stablehlo] lowering aten.randn & aten.normal_functional to mhlo.rng … (#3328) 2024-05-11 15:33:37 +08:00
Reduction.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
Rng.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
StablehloLegalizeUtils.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
TorchToStablehlo.cpp [Stablehlo] support uint8 (#3367) 2024-06-04 09:04:59 +08:00
Utils.cpp [Stablehlo] add torch_to_stablehlo::getBackendTypeForScalarType (#2975) 2024-03-04 23:31:54 +08:00
Utils.h [Stablehlo] add torch_to_stablehlo::getBackendTypeForScalarType (#2975) 2024-03-04 23:31:54 +08:00
ViewLike.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00