Skip to content

[TORCH] Support NLL loss with Gaussian distribution of target #4234

Open
@Prakhar-Dixit

Description

@Prakhar-Dixit

GaussianNLLLoss Function:

https://docs.pytorch.org/docs/stable/generated/torch.nn.GaussianNLLLoss.html#gaussiannllloss
Currently this op is not supported in torch-mlir.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions