openvinotoolkit
[JAX FE] Add translation for more operations.
Details:
- Add the translation for reduce_window_max, reduce_window_sum, rsqrt, reshape , squeeze, slice, broadcast_in_dim, copy, dot_general and transpose of JAX frontend
- Add corresponding test
Tickets:
- CVS-145575
- CVS-145583
- CVS-145580
- CVS-145574
- CVS-145581
- CVS-145579
- CVS-145582
- CVS-145573
- CVS-145578
NOTE: this PR should be merged after #25290
@mvafin Could you please help to review this PR? cc @rkazants
@mvafin Hi, could you please talk more about the implementation of copy translation? I tried to return the input directly but the test cannot be passed. Should I do something before the translation of it?
@mvafin Hi, could you please talk more about the implementation of
copytranslation? I tried to return the input directly but the test cannot be passed. Should I do something before the translation of it?
Just do the same as was done for pytorch FE: https://github.com/openvinotoolkit/openvino/blob/260cf42126ee322f49f5f8afc5d189d8055e98ba/src/frontends/pytorch/src/op_table.cpp#L424
No need to have an extra file for such operation, just create a common helper in utils.hpp and use it for copy and other similar operations.
