Searched refs:_simple_gather (Results 1 – 1 of 1) sorted by relevance
220 def _simple_gather(per_replica_value, reduce_to_device, axis): function614 gathered = _simple_gather(per_replica_value, reduce_to_device, axis)