mirror of https://github.com/llvm/torch-mlir
5928f68e60
* not to decompose `aten.amax` on `stablehlo` backend. Because it could be lowering to `stablehlo.reduce` directly. * lowering `aten.max.dim` to `stablehlo.reduce apply max` when `AtenMaxDimOp.getIndices()` doesn't have users. It's more simple. |
||
|---|---|---|
| .. | ||
| _torch_mlir_custom_op_example | ||
| csrc | ||
| jit_ir_importer | ||
| _dynamo_fx_importer.py | ||
| _version.py | ||
| dynamo.py | ||
| torchscript.py | ||