Home
last modified time | relevance | path

Searched refs:get_var_on_current_device (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dpacked_distributed_variable.py96 def get_var_on_current_device(self): member in PackedDistributedVariable
107 return self.get_var_on_current_device().handle
117 return self.get_var_on_current_device().value()
147 return update_fn(self.get_var_on_current_device(), value, **kwargs)
236 … return self.get_var_on_current_device()._dense_var_to_tensor( # pylint: disable=protected-access
287 return self._var.get_var_on_current_device().handle
Dpacked_distributed_variable_test.py53 self.assertAllEqual(packed_var.get_var_on_current_device(), v0)
58 self.assertAllEqual(packed_var.get_var_on_current_device(), v1)