Searched refs:_build_recursive_hd_gather (Results 1 – 1 of 1) sorted by relevance
471 reduced_shards = _build_recursive_hd_gather(input_tensors, devices, red_op)480 def _build_recursive_hd_gather(input_tensors, devices, red_op): function