xla
Fix bug in ShardTensor to generate correct shards for each device
#4549

Merged

Fix bug in ShardTensor to generate correct shards for each device #4549

JackCaoG merged 1 commit into master from jonbolin-sharding-util

jonb377 requested a review from

yeounoh 3 years ago

jonb377 force pushed from 6a1aed97 to 46df5c50 3 years ago

jonb377 force pushed from 46df5c50 to b6c31b30 3 years ago

jonb377 force pushed from b6c31b30 to 5853b48f 3 years ago

jonb377 force pushed from 5853b48f to 77a87f6b 3 years ago

Fix bug in ShardTensor to generate correct shards for each device

e6327440

jonb377 force pushed from 77a87f6b to e6327440 3 years ago

yeounoh commented on 2023-02-03

yeounoh approved these changes on 2023-02-03

JackCaoG merged a6a77ef8 into master 3 years ago

jonb377 deleted the jonbolin-sharding-util branch 3 years ago

Reviewers

yeounoh

Assignees

No one assigned

Labels

None yet

Milestone

No milestone