Searched refs:_apply_all_reduce (Results 1 – 1 of 1) sorted by relevance
47 return _apply_all_reduce('sum', tensors)93 return _apply_all_reduce('prod', tensors)110 return _apply_all_reduce('min', tensors)127 return _apply_all_reduce('max', tensors)210 def _apply_all_reduce(reduction, tensors): function