xla
Apply input sharding to all compatible input tensors
#4872

Merged

Apply input sharding to all compatible input tensors #4872

jonb377 merged 1 commit into master from jonbolin-generic-dataloader

jonb377 requested a review from

yeounoh 3 years ago

jonb377 requested a review from

steventk-g 3 years ago

jonb377 force pushed from 285ad26d to dfaeb87e 3 years ago

jonb377 changed the title ~~Allow sharding generic input formats~~ Apply input sharding to all compatible input tensors 3 years ago

JackCaoG commented on 2023-04-12

yeounoh requested changes on 2023-04-15

jonb377 force pushed from dfaeb87e to 710ea6be 3 years ago

Apply sharding to all compatible input tensors

41eba5ce

jonb377 force pushed from 710ea6be to 41eba5ce 3 years ago

yeounoh approved these changes on 2023-04-18

JackCaoG approved these changes on 2023-04-18

jonb377 merged 919f0aee into master 3 years ago

jonb377 deleted the jonbolin-generic-dataloader branch 3 years ago

Reviewers

JackCaoG

yeounoh

steventk-g

Assignees

No one assigned

Labels

None yet

Milestone

No milestone