Don't assign device for the keras part of _saved_first_checkpoint. Fix #14504.
authorYifei Feng <yifeif@google.com>
Wed, 21 Feb 2018 23:07:05 +0000 (15:07 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Wed, 21 Feb 2018 23:11:24 +0000 (15:11 -0800)
commit5e8aaa66af43b6b66e61ca7d589002eac6b4fb69
treea462c9222c0a11b3da08f9e74bf624056ce5f457
parent3c3da104f14709dd0495c5ae0783b69e7da21fb9
Don't assign device for the keras part of _saved_first_checkpoint. Fix #14504.

PiperOrigin-RevId: 186526175
tensorflow/python/keras/_impl/keras/estimator.py
tensorflow/python/keras/_impl/keras/estimator_test.py