Searched refs:_update_cross_replica (Results 1 – 1 of 1) sorted by relevance
70 return var._update_cross_replica(update_fn, v, **kwargs) # pylint: disable=protected-access865 def _update_cross_replica(self, update_fn, value, **kwargs): member in DistributedVariable925 return self._update_cross_replica(update_fn, value, **kwargs)