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/Dialect/Torch
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
..
IR Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00
Transforms [LINALG] Add decomposition of `aten.dropout` op 2022-03-22 13:14:49 +05:30
Utils Avoid `using` the `torch_upstream` namespace. 2022-03-15 17:24:17 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00