Searched refs:_simple_reduce (Results 1 – 2 of 2) sorted by relevance
200 def _simple_reduce(per_replica_value, reduce_to_device, accumulation_fn, function599 reduced = _simple_reduce(per_replica_value, reduce_to_device,
373 result = cross_device_ops_lib._simple_reduce(