Searched refs:_do_batch_all_reduce (Results 1 – 1 of 1) sorted by relevance
699 dense_results = self._do_batch_all_reduce(reduce_op, dense_values)710 def _do_batch_all_reduce(self, reduce_op, dense_values): member in AllReduceCrossDeviceOps