torch-mlir/lib/Dialect/Torch
Rob Suderman a78659742a
[onnx] Migrate `onnx.ReduceMax` to match `onnx.ReduceMin` (#2981)
This mostly copy-pastes the reduce minimum implementation to reduce max
to improve test coverage. We also improve the aten lowering for min/max
dim for unsigned types.
2024-03-06 16:48:21 -08:00
..
IR [torch] Additional folders for shape computations (#2972) 2024-03-04 11:46:49 -08:00
Transforms [onnx] Migrate `onnx.ReduceMax` to match `onnx.ReduceMin` (#2981) 2024-03-06 16:48:21 -08:00
Utils Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00