Searched defs:num_subchunks (Results 1 – 1 of 1) sorted by relevance
189 def _ring_permutations(num_workers, num_subchunks, gpu_perm): argument250 def build_ring_all_reduce(input_tensors, num_workers, num_subchunks, argument293 def _build_ring_gather(input_tensors, devices, num_subchunks, argument