Searched refs:_safe_mean (Results 1 – 3 of 3) sorted by relevance
150 def _safe_mean(losses, num_present): function181 loss = _safe_mean(loss, num_replicas * _num_elements(weighted_losses))
81 def _safe_mean(losses, num_present): function198 loss = _safe_mean(loss, denom)201 loss = _safe_mean(loss, num_replicas * _num_present(losses, weights))203 loss = _safe_mean(loss, num_replicas * _num_elements(losses))
69 def _safe_mean(losses, num_present): function118 mean_loss = _safe_mean(total_loss, num_present)