mirror of https://github.com/llvm/torch-mlir
7c3ba25238
- 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> |
||
|---|---|---|
| .. | ||
| test | ||
| torch_mlir/dialects | ||
| torch_mlir_e2e_test | ||
| CMakeLists.txt | ||
| TorchMLIRModule.cpp | ||