Fix multi-host ImageNet sharding when subset sizes are used#922

copybara-service[bot] merged 1 commit into

mastergoogle/init2winit:masterfrom

test_940692915google/init2winit:test_940692915Copy head branch name to clipboard