Searched refs:SetPSRankId (Results 1 – 3 of 3) sorted by relevance
72 void SetPSRankId(uint32_t rank_id);
141 void PSContext::SetPSRankId(uint32_t rank_id) { rank_id_ = rank_id; } in SetPSRankId() function in mindspore::ps::PSContext
38 PSContext::instance()->SetPSRankId(server_node_->rank_id()); in Run()