llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
824 Commits
45 Branches
1195 Tags
71 MiB
Commit Graph

3 Commits (d4ea39b616ff38162a340292e53928b645a6494b)

Author SHA1 Message Date
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
Anup Gangwar abd61b4974 * Workaround for Issue 521, remove createTosaToStandard from Passes.cpp and 
disable ElementwisePowModule_basic
* Update nll_loss_forward to align to the change in PyTorch

Signed-off-by: Anup Gangwar <anup.gangwar@arm.com>
 2022-01-12 14:30:58 -06:00
Prashant Kumar 977b1b03ea Add aten::nll_loss_forward op lowering. 
The op lowering has been added as a part of `torch-lower-to-linalg`
pass. This takes care of ignore_index but the weight and reduction
operand is still to be accounted for.

Signed-off-by: Prashant Kumar <prashant@nod-labs.com>
 2021-12-07 17:11:08 +05:30