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
History
Prashant Kumar 68acc8696e Modify softmax decomposition to be more numerically stable. 
The softmax decomposition is modified according to https://github.com/pytorch/functorch/blob/main/functorch/_src/decompositions.pytorch
to account for numerical stability. Also, modified aten.argmax lowering
to handle negative dimension.
 		2022-02-03 21:20:36 +05:30
..
Torch Modify softmax decomposition to be more numerically stable. 2022-02-03 21:20:36 +05:30
TorchConversion [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00