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
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
..
IR Fold __is__ and unchecked_cast of derefine 2022-01-28 17:54:40 -05:00
Transforms Add aten::nll_loss_backward op 2022-02-04 21:57:53 +05:30
Utils Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00