[Good First Issue]: Support aten::where(cond) for pytorch models

[mvafin]

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.

Currently this use case is unsupported: https://github.com/openvinotoolkit/openvino/blob/8d8d66ac1d60108934e031c0964ec242ace924d6/src/frontends/pytorch/src/op/where.cpp#L20

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.where 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-136483

[vdizhenin]

.take

[github-actions[bot]]

Thank you for looking into this issue! Please let us know if you have any questions or require any help.

[Aryan8912]

It's been two weeks since this issue was opened. Can you assign it to me?

[vdizhenin]

No, I'm still working on it

[p-wysocki]

Hello @vdizhenin, are you still working on that issue? Do you need any help?

[rishi-sangare]

.take

[github-actions[bot]]

Thank you for looking into this issue! Please let us know if you have any questions or require any help.