torch-mlir/lib/Dialect
Arham Khan 574143448b
[E2E][ONNX] torch.multinomial (#3404)
This PR adds a conversion in the TorchOnnxToTorch pass for the ONNX
Multinomial operation. It also adds a TorchToLinalg lowering for the
`aten.Multinomial` op and does a light refactor of some repeated code
that generates random floating point numbers in
`TorchToLinalg/Random.cpp`.
2024-07-16 23:09:39 +05:30
..
TMTensor [torch-mlir] bump to llvm/llvm-project@9b78ddf3b2 (#3491) 2024-06-27 19:28:02 -07:00
Torch [E2E][ONNX] torch.multinomial (#3404) 2024-07-16 23:09:39 +05:30
TorchConversion [Stablehlo] legalize deprecated ops to stablehlo ops (#3543) 2024-07-17 00:05:11 +08:00
CMakeLists.txt Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00