Searched defs:set_worker_num (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/core/ | ||
| D | node_manager.cc | 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 |
| D | abstract_node.cc | 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 |
| /third_party/mindspore/mindspore/ccsrc/ps/ | ||
| D | ps_context.cc | 254 void PSContext::set_worker_num(uint32_t worker_num) { in set_worker_num() function in mindspore::ps::PSContext |