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
..
CMakeLists.txt	Move external/torch-mlir to the root of the repo.	2021-09-27 17:11:08 -07:00
TorchToLinalg.cpp	Modify softmax decomposition to be more numerically stable.	2022-02-03 21:20:36 +05:30