Searched refs:gpu_perm (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce.py | 193 def _ring_permutations(num_workers, num_subchunks, gpu_perm): argument
223 num_gpus = len(gpu_perm)
236 default_order = [(w * num_gpus) + i for i in gpu_perm]
255 gpu_perm, red_op, un_op=None): argument
282 num_workers, num_subchunks, gpu_perm)
|
| D | all_reduce_test.py | 146 gpu_perm = range(0, num_gpus)
148 x, num_workers, subdiv, gpu_perm, math_ops.add, un_op)
|