Searched defs:gpu_perm (Results 1 – 1 of 1) sorted by relevance
189 def _ring_permutations(num_workers, num_subchunks, gpu_perm): argument251 gpu_perm, red_op, un_op=None): argument