llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch
History
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
..
IR [ONNX] Fix bug in ONNXToTorch PadOp's pads tensor rearrangement (#3485) 2024-07-03 15:02:49 -05:00
Transforms [E2E][ONNX] torch.multinomial (#3404) 2024-07-16 23:09:39 +05:30
Utils [TorchToLinalg] Fix Quantized Convolution Accumulator Type (#3459) 2024-06-20 13:54:20 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00