PR #6925 [SPMD] Support SPMDShardToFullShape

[SPMD] Support SPMDShardToFullShape #6925

alanwaketan merged 4 commits into master from alanwaketan/resume_spmd

initial commit

ad5b9874

minor fix

af57d908

Add an e2e test

14d32916

Fix linters

50f4c663

alanwaketan force pushed from 9c8643b2 to 50f4c663 2 years ago

alanwaketan requested a review from

yeounoh 2 years ago

alanwaketan requested a review from

jonb377 2 years ago

alanwaketan marked this pull request as ready for review 2 years ago

yeounoh commented on 2024-04-16

yeounoh approved these changes on 2024-04-16

alanwaketan merged 6e34bbed into master 2 years ago

alanwaketan deleted the alanwaketan/resume_spmd branch 2 years ago

Reviewers

yeounoh

jonb377

Assignees

No one assigned

Labels

None yet

Milestone

No milestone