Searched refs:_call_fn_for_each_replica (Results 1 – 1 of 1) sorted by relevance
1698 aggregated_output_loss = self._call_fn_for_each_replica(1958 return self._call_fn_for_each_replica(1966 def _call_fn_for_each_replica(self, fn, *args, **kwargs): member in Model