[TORCH] Support NLL loss with Poisson distribution of target

[zahidwx]

Poisson NLL loss function

https://docs.pytorch.org/docs/stable/generated/torch.nn.PoissonNLLLoss.html

Currently this op is not supported in torch-mlir. It can be lowered to existing primitive ops in torch dialect.

[Prakhar-Dixit]

I am interested in taking up this issue.
Could you please assign it to me ?
@zahidwx

[Prakhar-Dixit]

@vivekkhandelwal1

[vivekkhandelwal1]

Poisson NLL loss function

https://docs.pytorch.org/docs/stable/generated/torch.nn.PoissonNLLLoss.html

Currently this op is not supported in torch-mlir. It can be lowered to existing primitive ops in torch dialect.

@zahidwx, I hope you are not working on this issue.

[zahidwx]

Hi @Prakhar-Dixit, I was planning to take this but you can go ahead and take this issue no problem.

[zahidwx]

PR: #4232 merged to main.