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