Commit 97cba0b8 authored by Rohan Jain's avatar Rohan Jain Committed by TensorFlower Gardener
Browse files

Allowing source_device to be set to /cpu:0 for multi device iterator in... 

Allowing source_device to be set to /cpu:0 for multi device iterator in distribution strategies. That is always the appropriate option.

In the existing code, we would set it to a partially specified "worker" name that was ambiguous and end up on the GPU.

PiperOrigin-RevId: 214882658
parent 370d385c
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment