[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:

openvino/src/frontends/pytorch/src/op/where.cpp

Line 20 in 8d8d66a

PYTORCH_OP_CONVERSION_CHECK(!context.input_is_none(1), "aten::where(cond) unsupported");

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]

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]

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

[mlukasze]

hey @rishi-sangare any help needed?

[gatbontonp]

Hi! I'm relatively new to open source. Can I try and work this?

[gatbontonp]

.take

[github-actions]

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

[p-wysocki]

@gatbontonp Hi, thanks for taking the issue! Please let us know if you have any questions or need help. :)