torch-mlir/lib/Conversion/TorchToStablehlo
Rob Suderman 2374b9e02d
Bump to llvm/llvm-project@e813750354 (#3765)
Includes stablehlo bump
2024-10-04 12:08:35 -07:00
..
Basic.cpp [Stablehlo] fix aten compare ops' promote rules (#3709) 2024-09-13 18:48:41 +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] use stablehlo specs lowering AtenSliceScatterOp (#3592) 2024-08-15 20:06:29 +08:00
Linear.cpp Bump to llvm/llvm-project@e813750354 (#3765) 2024-10-04 12:08:35 -07:00
Pooling.cpp [stablehlo] support aten_adaptive_max_pool1d lowering (#3728) 2024-09-26 11:42:38 +08:00
PopulatePatterns.h [Stablehlo] lowering aten.randn & aten.normal_functional to mhlo.rng … (#3328) 2024-05-11 15:33:37 +08:00
Reduction.cpp [Stablehlo] support aten.all.dim (#3746) 2024-09-30 15:59:27 +08:00
Rng.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
StablehloLegalizeUtils.cpp [stablehlo] support dynamic-shaped index in stablehlo conversion for aten.index-like ops (#3322) 2024-08-01 10:41:09 +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 [Stablehlo] use index type as dim size, avoid to generate index_cast (#3526) 2024-07-07 18:03:03 +08:00