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/Conversion
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
..
TorchToLinalg Modify softmax decomposition to be more numerically stable. 2022-02-03 21:20:36 +05:30
TorchToSCF Update external/llvm-project 2022-01-18 13:25:42 -05:00
TorchToStd Update external/llvm-project 2022-01-18 13:25:42 -05:00
TorchToTosa [tosa] Add maxpool2d and adaptive_avgpool2d support (#550) 2022-01-31 13:34:09 -08:00
CMakeLists.txt Add a basic TOSA E2E backend. 2021-10-08 09:59:45 -07:00
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp Add a basic TOSA E2E backend. 2021-10-08 09:59:45 -07:00