torch-mlir/python
Prashant Kumar 102c497c4c Add decomposition of _log_softmax op.
Decompose _log_softmax into log(softmax(x)).
2022-02-10 23:17:26 +05:30
..
test Make error reporting a bit better. 2021-10-28 13:20:16 -07:00
torch_mlir/dialects Add decomposition of _log_softmax op. 2022-02-10 23:17:26 +05:30
torch_mlir_e2e_test Include IR dump options on e2e failure report 2022-02-09 11:19:34 -05:00
CMakeLists.txt Enable python extensions when building out of tree (#363) 2021-10-27 17:04:12 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00