Searched refs:_num_elements (Results 1 – 5 of 5) sorted by relevance
250 def _num_elements(losses): function264 loss = _safe_mean(loss, _num_elements(weighted_losses))
116 return _num_elements(losses)133 def _num_elements(losses): function200 loss = _safe_mean(loss, _num_elements(losses))
647 def _num_elements(grad): function713 num_elements_fn=_num_elements,
96 if value._num_elements() > self._max_tensor_size: # pylint: disable=protected-access
1127 def _num_elements(self): member in _EagerTensorBase