openvinotoolkit/openvino

[Good First Issue]: Support aten::rrelu and aten::rrelu_ for pytorch models

Open

#23.746 geöffnet am 28. März 2024

Auf GitHub ansehen
 (28 Kommentare) (0 Reaktionen) (1 zugewiesene Person)C++ (3.229 Forks)auto 404
category: PyTorch FEgood first issueno_stale

Repository-Metriken

Stars
 (10.286 Stars)
PR-Merge-Metriken
 (Durchschn. Merge 14T 3h) (305 gemergte PRs in 30 T)

Beschreibung

Context

OpenVINO component responsible for support of PyTorch models is called as PyTorch Frontend (PT FE). PT FE converts a model represented as TorchScript model to a model in OpenVINO opset.

This is a randomized operation, but it is randomized only when training, for OpenVINO we assume that models are in inference only mode, so randomization is not needed. Due to that this op is actually similar to torch.nn.LeakyReLU

What needs to be done?

  • Implement conversion rule and/or transformation to support the new operation.
  • Implement operation tests in tests/layer_tests/pytorch_tests. Please consider different data types, but keep reasonable number of test cases

Example Pull Requests

#18998

Resources

Contact points

@openvinotoolkit/openvino-pytorch-frontend-maintainers

Ticket

CVS-136484

Contributor Guide