Skip to content

[TF FE] Support complex tensors for Gather, GatherV2 operations #29984

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
wants to merge 5 commits into
base: master
Choose a base branch
from
Open

[TF FE] Support complex tensors for Gather, GatherV2 operations #29984

wants to merge 5 commits into from
+110 −13

Conversation

JatinSachdeva2004
Copy link

@rkazants @p-wysocki

Changes

  • Propagated ComplexTypeMark from input to output in GatherV2 loader.
  • Removed redundant ComplexTypeMark from output (direct floating-point representation).
  • Updated axis calculation logic for complex tensors.
  • Added layer tests (test_tf_Gather.py) for complex tensor validation.
  • Temporarily skipped GPU tests (investigation pending).

Verification

  • All tf_Gather tests pass locally on CPU: 
    TEST_DEVICE=CPU pytest test_tf_Gather.py

Related Issues
Fixes [#22951]

Linked to PR [#29346]

@JatinSachdeva2004 JatinSachdeva2004 requested a review from a team as a code owner April 7, 2025 21:47
@github-actions github-actions bot added category: TF FE OpenVINO TensorFlow FrontEnd category: TFL FE OpenVINO TensorFlow Lite FrontEnd labels Apr 7, 2025
@sys-openvino-ci sys-openvino-ci added the ExternalPR External contributor label Apr 7, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
category: TF FE OpenVINO TensorFlow FrontEnd category: TFL FE OpenVINO TensorFlow Lite FrontEnd ExternalPR External contributor
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@JatinSachdeva2004 @sys-openvino-ci @NingLi670