Searched refs:_build_shuffle_gather (Results 1 – 1 of 1) sorted by relevance
578 reduced_shards = _build_shuffle_gather(input_tensors, gather_devices,586 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): function824 reduced_shards = _build_shuffle_gather(