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