Searched defs:num_subchunks (Results 1 – 1 of 1) sorted by relevance
193 def _ring_permutations(num_workers, num_subchunks, gpu_perm): argument254 def build_ring_all_reduce(input_tensors, num_workers, num_subchunks, argument297 def _build_ring_gather(input_tensors, devices, num_subchunks, argument