-
Notifications
You must be signed in to change notification settings - Fork 74k
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.
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
[TF:TRT] Support GatherV2 op when indices is constant #52268
[TF:TRT] Support GatherV2 op when indices is constant #52268
Conversation
Thanks for your pull request. It looks like this may be your first contribution to a Google open source project (if not, look below for help). Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA). 📝 Please visit https://cla.developers.google.com/ to sign. Once you've signed (or fixed any issues), please reply here with What to do if you already signed the CLAIndividual signers
Corporate signers
ℹ️ Googlers: Go here for more info. |
@googlebot I signed it! |
In your PR description, you have: we actually require batch size 1 when (1) both input and indices are tensors, or (2) both input and indices are constants. |
@bixia1 , regarding your comment:
Thanks for pointing this out.
I could remove my note on the PR, as it sounds redundant. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me. Thanks for your work!
4891e03
to
7cc5fbd
Compare
556010c
to
7a9ddb4
Compare
Looks good to me. I modified the PR description, please check. |
Convert 'indices' input to a GatherV2 op into a constant layer if it is a constant. Modify the unit tests to generate constant indices. Signed-off-by: Meenakshi Venkataraman <meenakshiv@nvidia.com>
@bixia1 -- squash done. |
7a9ddb4
to
9b8fe6c
Compare
I am working on fixing some minor format issues and merging this PR from my end. |
Convert the 'indices' input to a GatherV2 op into a constant layer when 'indices' is a constant.
Modify the unit tests to generate constant indices.
Signed-off-by: Meenakshi Venkataraman meenakshiv@nvidia.com
Tagging @bixia1 for review
CC: @tfeher @DEKHTIARJonathan