Searched refs:get_var_on_current_device (Results 1 – 2 of 2) sorted by relevance
96 def get_var_on_current_device(self): member in PackedDistributedVariable107 return self.get_var_on_current_device().handle117 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-access287 return self._var.get_var_on_current_device().handle
53 self.assertAllEqual(packed_var.get_var_on_current_device(), v0)58 self.assertAllEqual(packed_var.get_var_on_current_device(), v1)