Commit 69ac7077 authored by (David) Siu-Kei Muk's avatar (David) Siu-Kei Muk
Browse files

1. Moved estimator._device_fn to RunConfig as @property 

2. Made RunConfig.device_fn to return custom device function if one is specified, otherwise the result from `tf.train.replica_device_setter` call is used
3. Added some basic unit tests, may need further tests.
parent 4b98b9db
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