Searched refs:_get_on_device_or_primary (Results 1 – 3 of 3) sorted by relevance
49 …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 DistributedValues412 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_value632 return self._get_on_device_or_primary().eval(session)647 return self._get_on_device_or_primary().device695 def _get_on_device_or_primary(self): member in DistributedVariable721 return self._get_on_device_or_primary().value()1072 return self._get_on_device_or_primary()._as_graph_element() # pylint: disable=protected-access[all …]
116 var = self._get_on_device_or_primary()143 with ops.device(self._get_on_device_or_primary().device):
770 slot = slot._get_on_device_or_primary() # pylint: disable=protected-access