xla
[SPMD] Use on_device_shape in ExecuteReplicated
#4139

Merged

[SPMD] Use on_device_shape in ExecuteReplicated #4139

yeounoh merged 1 commit into master from spmd_bug_fix

[SPMD] Use on_device_shape in ExecuteReplicated

9c308bcd

yeounoh requested a review from

jonb377 3 years ago

yeounoh requested a review from

JackCaoG 3 years ago

yeounoh assigned

yeounoh 3 years ago

jonb377 approved these changes on 2022-10-28

yeounoh merged 8f3edf45 into master 3 years ago

Reviewers

jonb377

JackCaoG

Assignees

yeounoh

Labels

None yet

Milestone

No milestone