Searched refs:_build_shuffle_gather (Results 1 – 1 of 1) sorted by relevance
584 reduced_shards = _build_shuffle_gather(input_tensors, gather_devices,592 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): function830 reduced_shards = _build_shuffle_gather(