Searched refs:_batch_reduce (Results 1 – 1 of 1) sorted by relevance
559 def _batch_reduce(test_object, reduce_op, value_destination_pairs): function626 _batch_reduce(collective_all_reduce, reduce_util.ReduceOp.MEAN,631 _batch_reduce(collective_all_reduce, reduce_util.ReduceOp.SUM,