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

2 Commits (454fa9d12338b289805d295a934a076aaf460e88)

Author SHA1 Message Date
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