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
..
torchscript	Modify softmax decomposition to be more numerically stable.	2022-02-03 21:20:36 +05:30