xla
Add test to train sharded resnet50 model
#4141
Merged

Loading