Lines Matching refs:peer_rank

9     def __init__(self, comm, peer_rank, peer_tag):  argument
12         self.peer_rank = peer_rank
19             self.peer_rank)
31         self.comm.send(xs, self.peer_rank, self.peer_tag)
39         grad = self.comm.recv(self.peer_rank, self.peer_tag)
49     def __init__(self, comm, peer_rank, peer_tag):  argument
52         self.peer_rank = peer_rank
72             self.peer_rank)
75         data = self.comm.recv(self.peer_rank, self.peer_tag)
84         self.comm.send(grad_outputs, self.peer_rank, self.peer_tag)
136             communicator, peer_rank=rank, peer_tag=tag)(*inputs)
139             communicator, peer_rank=rank, peer_tag=tag)(x, dummy_var)
189             peer_rank=rank,
195             peer_rank=rank,