xla
Apply sharding when creating tensors data in data loader
#5015
Merged

Apply sharding when creating tensors data in data loader #5015

jonb377 merged 2 commits into master from jonbolin-pybind-dl
jonb377
jonb377 Revert "Revert "Apply sharding when creating tensors data in data loa…
b3839b89
jonb377 jonb377 added distributed
jonb377 jonb377 requested a review from yeounoh yeounoh 2 years ago
jonb377 Change test tensor shape to support v3
0b993b5d
jonb377 jonb377 force pushed from 2f42e97b to 0b993b5d 2 years ago
yeounoh
yeounoh approved these changes on 2023-05-16
jonb377 jonb377 merged f34f0b0a into master 2 years ago
jonb377 jonb377 deleted the jonbolin-pybind-dl branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone