Home
last modified time | relevance | path

Searched refs:_get_on_device_or_primary (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dvalues.py49 …return update_fn(var._get_on_device_or_primary(), value, **kwargs) # pylint: disable=protected-ac…
161 def _get_on_device_or_primary(self): member in DistributedValues
412 return self._get_on_device_or_primary()
565 return self._get_on_device_or_primary().initialized_value()
569 return self._get_on_device_or_primary().initial_value
632 return self._get_on_device_or_primary().eval(session)
647 return self._get_on_device_or_primary().device
695 def _get_on_device_or_primary(self): member in DistributedVariable
721 return self._get_on_device_or_primary().value()
1072 return self._get_on_device_or_primary()._as_graph_element() # pylint: disable=protected-access
[all …]
Dtpu_values.py116 var = self._get_on_device_or_primary()
143 with ops.device(self._get_on_device_or_primary().device):
/external/tensorflow/tensorflow/python/training/
Doptimizer.py770 slot = slot._get_on_device_or_primary() # pylint: disable=protected-access