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
Gaurav Shukla 7c3ba25238 [LINALG] Add decomposition of `aten.dropout` op 
- This commit adds decomposition of `aten.dropout` op. It also covers the
  training mode of the same op.
- It also adds lowering of `aten.sub.float` op.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 		2022-03-22 13:14:49 +05:30
..
TorchToLinalg [MLIR][TORCH] Add value tensor variant to aten::copy_ op 2022-03-21 19:12:37 +05:30
TorchToSCF Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
TorchToStd [LINALG] Add decomposition of `aten.dropout` op 2022-03-22 13:14:49 +05:30
TorchToTMTensor Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
TorchToTosa [tosa] Support for Aten[Zeros|Ones|Fill_Scalar] ops (#604) 2022-02-16 09:53:51 -08:00
Utils [MLIR][TORCH] Move common helper functions to Utils.cpp 2022-03-08 22:52:34 +05:30
CMakeLists.txt [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30