llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/projects/pt1/python/torch_mlir
History
Yuanqiang Liu 5bb1a65ec9
[Stablehlo] refactor reduction lowering and support aten.amin (#3383) 
* implement detailed lowering template pattern
`ConvertAtenReduceAllDimsOp` and `ConvertAtenReduceKeepDimOp`
* support `aten.amin`'s lowering.
 		2024-05-23 20:40:20 +08:00
..
_torch_mlir_custom_op_example [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
csrc [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
jit_ir_importer [Stablehlo] refactor reduction lowering and support aten.amin (#3383) 2024-05-23 20:40:20 +08:00
_dynamo_fx_importer.py [NFC] Update black version (#3256) 2024-04-29 11:06:01 +08:00
_version.py [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
dynamo.py [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
torchscript.py [Stablehlo] refactor reduction lowering and support aten.amin (#3383) 2024-05-23 20:40:20 +08:00