llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test
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
..
Conversion * [tosa] Support for AtenFlattenUsingIntsOp (#548) 2022-01-28 21:38:56 -08:00
Dialect Modify softmax decomposition to be more numerically stable. 2022-02-03 21:20:36 +05:30
RefBackend [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
python Bring some more order to the e2e error reporting situation. 2021-10-08 13:00:12 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
lit.cfg.py Add tool substitutions to support out-of-tree builds 2021-10-07 21:16:43 -07:00
lit.site.cfg.py.in Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00