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