llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python
History
Prashant Kumar ccf546f14c Add aten::nll_loss_backward op 
The lowering of aten::nll_loss_backward op has been added
from torch to linalg dialect. The changes has been made as
a part of -torch-convert-to-linalg pass.

Signed-off-by: Prashant Kumar prashant@nod-labs.com
 		2022-02-04 21:57:53 +05:30
..
test Make error reporting a bit better. 2021-10-28 13:20:16 -07:00
torch_mlir/dialects Add aten::nll_loss_backward op 2022-02-04 21:57:53 +05:30
torch_mlir_e2e_test [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -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