| D | cross_device_ops_test.py | 113 def __init__(self, num_processes): argument
131 def get_global_mpr(num_processes): argument
152 def make_collective(self, num_processes, gpu_per_process): argument
325 def testReduceDense(self, num_processes, required_gpus, implementation, argument
372 def testReduceSparse(self, num_processes, required_gpus, implementation, argument
457 def testBatchReduceDense(self, num_processes, required_gpus, implementation, argument
505 def testBatchReduceSparse(self, num_processes, required_gpus, implementation, argument
634 def testAllReduceDense(self, num_processes, required_gpus, implementation, argument
702 def testAllReduceSparse(self, num_processes, required_gpus, implementation, argument
781 def testAllReduceMixedDenseAndSparse(self, num_processes, required_gpus, argument
[all …]
|