torch-mlir/lib/Dialect/Torch
Xinyu Yang f32ada993d
[Stablehlo] Improve the lowering of pool op in stablehlo (#3259)
1. Handle case stride == None
2. add avgpool3d maxpool1d  maxpool3d lowering
2024-05-01 00:06:13 +08:00
..
IR [ONNX] Fix Onnx.Selu lowering and canonicalizer for IntImplicit op (#3221) 2024-04-29 04:00:01 +00:00
Transforms [Stablehlo] Improve the lowering of pool op in stablehlo (#3259) 2024-05-01 00:06:13 +08:00
Utils Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00