Bug fix: outfeeds have to be placed on host devices. The previous version works when there's only 1 host, as in the tests, but fails when using larger slices.
Bug fix: outfeeds have to be placed on host devices. The previous version works when there's only 1 host, as in the tests, but fails when using larger slices.

PiperOrigin-RevId: 232807234

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo