Commit 78af5c7e authored by Igor Saprykin's avatar Igor Saprykin Committed by TensorFlower Gardener
Browse files

Correctly treat "devices=/gpu:0" argument of replicate_model_fn. 

At the moment if "devices=/GPU:0" are specified by the user, then variables are going to be placed on the GPU.  However, if "devices=/gpu:0" are given, then they are going to be placed on the CPU.  Instead, the latter case should be equivalent to the former case.

PiperOrigin-RevId: 184612823
parent 238bae44
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