experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix TPUMirroredVariable.device when called within a TPU context.

Browse Source 
PiperOrigin-RevId: 236814142
This commit is contained in:
Chris Jones 2019-03-05 02:37:31 -08:00 committed by TensorFlower Gardener
parent 5456cc28f3
commit 55cd3abd02
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/distribute/values.py
View File

@ -904,7 +904,7 @@ class TPUMirroredVariable(trackable.Trackable):
@property
def device(self):
return self._get().device
return self.handle.device
def eval(self, session=None):
return self.primary.eval(session)