Searched refs:set_worker_num (Results 1 – 11 of 11) sorted by relevance
30 node_->set_worker_num(worker_num); in Recover()32 node_->set_worker_num(PSContext::instance()->cluster_config().initial_worker_num); in Recover()
30 scale_out_message.set_worker_num(manager.worker_num()); in ScaleOutAsync()50 scale_in_message.set_worker_num(manager.worker_num()); in ScaleInAsync()
96 void set_worker_num(const int32_t &worker_num);
348 send_metadata_message.set_worker_num(node_manager_.worker_num()); in SendMetadata()579 node_manager_.set_worker_num(total_worker_num); in ProcessScaleOut()662 node_manager_.set_worker_num(total_worker_num); in ProcessScaleIn()
121 void set_worker_num(const int32_t &worker_num);
304 void NodeManager::set_worker_num(const int32_t &worker_num) { meta_data_->worker_num = IntToUint(wo… in set_worker_num() function in mindspore::ps::core::NodeManager
434 void AbstractNode::set_worker_num(const int32_t &worker_num) { worker_num_ = worker_num; } in set_worker_num() function in mindspore::ps::core::AbstractNode
52 "worker_num": ps_context().set_worker_num,
98 void set_worker_num(uint32_t worker_num);
254 void PSContext::set_worker_num(uint32_t worker_num) { in set_worker_num() function in mindspore::ps::PSContext
365 .def("set_worker_num", &PSContext::set_worker_num, "Set worker number.") in PYBIND11_MODULE()