torch-mlir

Commit Graph

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

Author

SHA1

Message

Date

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

1 Commits (396ab35c9d29b14dc2a55406edaa29d06a8c68fc)