Searched refs:_build_recursive_hd_scatter (Results 1 – 1 of 1) sorted by relevance
468 output_tensors = _build_recursive_hd_scatter(reduced_shards, devices)515 def _build_recursive_hd_scatter(input_tensors, devices): function