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