Searched refs:server_rank (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/worker/ |
| D | fl_worker.cc | 98 bool FLWorker::SendToServer(uint32_t server_rank, const void *data, size_t size, ps::core::TcpUserC… in SendToServer() argument
122 …if (!worker_node_->Send(ps::core::NodeRole::SERVER, server_rank, message, size, static_cast<int>(c… in SendToServer()
124 MS_LOG(ERROR) << "Sending message to server " << server_rank << " failed."; in SendToServer()
128 MS_LOG(WARNING) << "Response from server " << server_rank << " is empty."; in SendToServer()
134 MS_LOG(INFO) << "The server " << server_rank << " is in safemode or finished."; in SendToServer()
141 …if (!worker_node_->Send(ps::core::NodeRole::SERVER, server_rank, message, size, static_cast<int>(c… in SendToServer()
143 MS_LOG(ERROR) << "Sending message to server " << server_rank << " failed."; in SendToServer()
|
| D | fl_worker.h | 70 …bool SendToServer(uint32_t server_rank, const void *data, size_t size, ps::core::TcpUserCommand co…
|
| /third_party/mindspore/mindspore/boost/ |
| D | boost_cell_wrapper.py | 177 self.server_rank = _rank % _device_number
183 current_weights = self.weights[self.start[self.server_rank]: self.end[self.server_rank]]
247 rank_weights = self.weights[self.start[self.server_rank]: self.end[self.server_rank]]
257 if i == self.server_rank:
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | iteration.h | 88 bool ReInitForScaling(uint32_t server_num, uint32_t server_rank);
|
| D | iteration.cc | 180 bool Iteration::ReInitForScaling(uint32_t server_num, uint32_t server_rank) { in ReInitForScaling() argument
187 if (server_rank != kLeaderServerRank) { in ReInitForScaling()
188 if (!SyncIteration(server_rank)) { in ReInitForScaling()
|