Skip to content
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

[ONNX] Update constant-folding of Gather op #50554

Merged
merged 9 commits into from Jan 28, 2021

Conversation

KsenijaS
Copy link
Contributor

Update constant-folding of Gather operator so it also includes cases where rank of indices input is 0.
Currently it only support cases where rank of indices is 1.

@facebook-github-bot facebook-github-bot added cla signed oncall: jit Add this issue/PR to JIT oncall triage queue labels Jan 14, 2021
@codecov
Copy link

codecov bot commented Jan 28, 2021

Codecov Report

Merging #50554 (dd1d590) into onnx_ms_1 (cc211bb) will decrease coverage by 0.00%.
The diff coverage is 85.18%.

@@              Coverage Diff              @@
##           onnx_ms_1   #50554      +/-   ##
=============================================
- Coverage      80.85%   80.84%   -0.01%     
=============================================
  Files           1931     1931              
  Lines         210932   210959      +27     
=============================================
+ Hits          170542   170560      +18     
- Misses         40390    40399       +9     

BowenBao added a commit that referenced this pull request Feb 2, 2021
Update constant-folding of Gather operator so it also includes cases where rank of indices input is 0.
Currently it only support cases where rank of indices is 1.

Differential Revision: [D26191323](https://our.internmc.facebook.com/intern/diff/D26191323)

[ghstack-poisoned]
BowenBao added a commit that referenced this pull request Feb 3, 2021
Update constant-folding of Gather operator so it also includes cases where rank of indices input is 0.
Currently it only support cases where rank of indices is 1.

Differential Revision: [D26191323](https://our.internmc.facebook.com/intern/diff/D26191323)

[ghstack-poisoned]
facebook-github-bot pushed a commit that referenced this pull request Feb 4, 2021
Summary:
Pull Request resolved: #51514

Update constant-folding of Gather operator so it also includes cases where rank of indices input is 0.
Currently it only support cases where rank of indices is 1.

Test Plan: Imported from OSS

Reviewed By: pbelevich

Differential Revision: D26191323

Pulled By: SplitInfinity

fbshipit-source-id: 7edcbd8835b0248fefb908aca394f5cca5eae29e
BowenBao added a commit to BowenBao/pytorch that referenced this pull request Feb 5, 2021
Update constant-folding of Gather operator so it also includes cases where rank of indices input is 0.
Currently it only support cases where rank of indices is 1.

ghstack-source-id: 3752c5ada1735d17b6b785cc274b9169145e94e5
Pull Request resolved: pytorch#51514
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla signed oncall: jit Add this issue/PR to JIT oncall triage queue open source
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

6 participants