openvinotoolkit/openvino
Voir sur GitHub[Good First Issue]: Support aten::rrelu and aten::rrelu_ for pytorch models
Open
#23 746 ouverte le 28 mars 2024
category: PyTorch FEgood first issueno_stale
Métriques du dépôt
- Stars
- (10 286 stars)
- Métriques de merge PR
- (Merge moyen 14j 3h) (305 PRs mergées en 30 j)
Description
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
- torch.nn.RReLU description
- Contribution guide - start here!
- What is OpenVINO?
- OpenVINO PyTorch Frontend
- Blog post on contributing to OpenVINO
- User documentation
Contact points
@openvinotoolkit/openvino-pytorch-frontend-maintainers
Ticket
CVS-136484