torch-mlir/lib/Conversion/TorchToStablehlo
Xinyu Yang 23b53050de
[Torch]Support conv_transpose1d and conv_transpose3d (#3286)
1. Support conv_transpose1d and conv_transpose3d
2. Fix bugs of convertTransposedConv func in
lib/Conversion/TorchToStablehlo/Linear.cpp
2024-06-03 15:11:12 +08:00
..
Basic.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
CMakeLists.txt [Stablehlo] lowering aten.randn & aten.normal_functional to mhlo.rng … (#3328) 2024-05-11 15:33:37 +08:00
GatherScatter.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07: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 [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +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