torch-mlir

Author	SHA1	Message	Date
Gaurav Shukla	dcef4751f9	[LINALG] Fix name conflict of `self` keyword. - The `self` name is being used as a keyword argument to the `torch.ops.aten.nll_loss_backward` function call, which produces name-conflict error with the python keyword `self` which is pointer to the current object. - This commit fixes this issue by replacing the keyword argument by positional argument. Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>	2022-02-14 22:55:31 +05:30
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